57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
|
import os
|
||
|
|
import cv2
|
||
|
|
|
||
|
|
# ----------------------------
|
||
|
|
# 配置
|
||
|
|
# ----------------------------
|
||
|
|
SOURCE_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/61/1" # 原始图片根目录
|
||
|
|
TARGET_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/1" # 输出根目录
|
||
|
|
CLASSES = ["class0","class1","class2"] # 类
|
||
|
|
TARGET_SIZE = 640 # resize 尺
|
||
|
|
SUBSETS = ["train", "val", "test"]
|
||
|
|
|
||
|
|
# ----------------------------
|
||
|
|
# 全局 ROI (x, y, w, h)
|
||
|
|
# ----------------------------
|
||
|
|
GLOBAL_ROI = [2,880,385,200]
|
||
|
|
# ----------------------------
|
||
|
|
# 主处理函数
|
||
|
|
# ----------------------------
|
||
|
|
def process_images():
|
||
|
|
x, y, w, h = GLOBAL_ROI
|
||
|
|
for subset in SUBSETS:
|
||
|
|
for class_dir in CLASSES:
|
||
|
|
src_dir = os.path.join(SOURCE_ROOT_DIR, subset, class_dir)
|
||
|
|
tgt_dir = os.path.join(TARGET_ROOT_DIR, subset, class_dir)
|
||
|
|
os.makedirs(tgt_dir, exist_ok=True)
|
||
|
|
|
||
|
|
if not os.path.exists(src_dir):
|
||
|
|
print(f"警告: 源目录 {src_dir} 不存在，跳过")
|
||
|
|
continue
|
||
|
|
|
||
|
|
for file in os.listdir(src_dir):
|
||
|
|
if not (file.endswith(".jpg") or file.endswith(".png")):
|
||
|
|
continue
|
||
|
|
|
||
|
|
img_path = os.path.join(src_dir, file)
|
||
|
|
img = cv2.imread(img_path)
|
||
|
|
if img is None:
|
||
|
|
print(f"❌ 无法读取图片: {img_path}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
h_img, w_img = img.shape[:2]
|
||
|
|
x1, y1 = max(0, x), max(0, y)
|
||
|
|
x2, y2 = min(w_img, x + w), min(h_img, y + h)
|
||
|
|
|
||
|
|
cropped = img[y1:y2, x1:x2]
|
||
|
|
if cropped.size == 0:
|
||
|
|
print(f"❌ 裁剪结果为空: {file}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
resized = cv2.resize(cropped, (TARGET_SIZE, TARGET_SIZE))
|
||
|
|
tgt_path = os.path.join(tgt_dir, file)
|
||
|
|
cv2.imwrite(tgt_path, resized)
|
||
|
|
print(f"✅ 图片处理完成: {subset}/{class_dir}/{file}")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
process_images()
|