mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-07-26 18:30:12 +00:00
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
|
import os
|
|||
|
|
import cv2
|
|||
|
|
|
|||
|
|
dataset_path = 'datasets/v4_4_test_dataset'
|
|||
|
|
annotation_file = 'datasets/v4_4_test_dataset/label.txt'
|
|||
|
|
|
|||
|
|
small_images_path = 'datasets/v4_4_test_dataset_small'
|
|||
|
|
new_annotation_file = 'datasets/v4_4_test_dataset_small/label.txt'
|
|||
|
|
|
|||
|
|
os.makedirs(small_images_path, exist_ok=True)
|
|||
|
|
|
|||
|
|
with open(annotation_file, 'r') as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
|
|||
|
|
for i, line in enumerate(lines):
|
|||
|
|
image_name = line.split(" ")[0]
|
|||
|
|
|
|||
|
|
image_path = os.path.join(dataset_path, image_name)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
image = cv2.imread(image_path)
|
|||
|
|
height, width, _ = image.shape
|
|||
|
|
|
|||
|
|
# 如果图像的宽度和高度都小于2000而且长宽比小于2，将其复制到新的文件夹，并保存其标注信息
|
|||
|
|
if height < 2000 and width < 2000:
|
|||
|
|
if max(height, width)/min(height,width) < 2:
|
|||
|
|
print(i, height, width, image_path)
|
|||
|
|
small_image_path = os.path.join(small_images_path, image_name)
|
|||
|
|
cv2.imwrite(small_image_path, image)
|
|||
|
|
with open(new_annotation_file, 'a') as f:
|
|||
|
|
f.write(f'{line}')
|
|||
|
|
except:
|
|||
|
|
continue
|