From 410ac16efdf935c257de67b5372104a8f552587a Mon Sep 17 00:00:00 2001 From: "lamonov.pavel" Date: Tue, 8 Jul 2025 08:30:17 +0300 Subject: [PATCH] =?UTF-8?q?refs=20#=20=D0=9C=D0=B0=D1=81=D1=88=D1=82=D0=B0?= =?UTF-8?q?=D0=B1=D0=B8=D1=80=D0=BE=D0=B2=D0=B0=D0=BD=D0=B8=D0=B5=20=D0=B8?= =?UTF-8?q?=D0=B7=D0=BE=D0=B1=D1=80=D0=B0=D0=B6=D0=B5=D0=BD=D0=B8=D1=8F=20?= =?UTF-8?q?=D0=B4=D0=BE=20=D0=BE=D0=BF=D1=80=D0=B5=D0=B4=D0=B5=D0=BB=D1=91?= =?UTF-8?q?=D0=BD=D0=BD=D0=BE=D0=B9=20=D0=B2=D1=8B=D1=81=D0=BE=D1=82=D1=8B?= =?UTF-8?q?.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/dataset_validation.py | 26 ++++++++++++++++++++++++-- 1 file changed, 24 insertions(+), 2 deletions(-) diff --git a/scripts/dataset_validation.py b/scripts/dataset_validation.py index 7250c24..7cc6a9e 100644 --- a/scripts/dataset_validation.py +++ b/scripts/dataset_validation.py @@ -1,5 +1,6 @@ import os import cv2 +from PIL import Image def rename_files(directory_path: str): extensions = set() @@ -37,8 +38,29 @@ def check_symbols(dir_path: str): if any([char not in dict_chars for char in label]): print(filename) +def max_height(dir_path: str): + max_height = 0 + max_filename = '' + for filename in os.listdir(dir_path): + im = Image.open(os.path.join(dir_path, filename)) + if im.height > max_height: + max_height = im.height + max_filename = filename + + print(max_filename, max_height) + +def resize_to_height(dir_path: str, target_height=48): + for filename in os.listdir(dir_path): + with Image.open(os.path.join(dir_path, filename)) as img: + width_percent = target_height / float(img.height) + new_width = int(float(img.width) * width_percent) + resized_img = img.resize((new_width, target_height), Image.LANZOS) + resized_img.save(os.path.join(dir_path, filename)) + + + # rename_files("train_data/images/") # check_images("train_data/images/") # check_labels("train_data/images/") -check_symbols("train_data/") - +# check_symbols("train_data/") +# max_height("train_data/images")