Skip to content

Commit

Permalink
❇️ Add check_image_size.py
Browse files Browse the repository at this point in the history
  • Loading branch information
nanpuhaha committed Feb 10, 2023
1 parent c6d99c3 commit 898e0d0
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions utils/check_image_size.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
from collections import Counter
from pathlib import Path

from PIL import Image
from rich.console import Console

INPUT_DIR = Path("/opt/ml/input")
AIHUB_DIR = INPUT_DIR / "aihub"
CAR_DIR = INPUT_DIR / "car"
ANN_DIR = CAR_DIR / "annotations"
IMG_DIR = CAR_DIR / "images"


if __name__ == "__main__":
console = Console()
sizes = []
for _path in IMG_DIR.glob("*.jpg"):
img = Image.open(_path)
sizes.append((img.width, img.height))

console.print(Counter(sizes))
# Counter({(1920, 1080): 13313})

0 comments on commit 898e0d0

Please sign in to comment.