图像和对应标签重命名

图像和对应标签重命名

Version 1

import cv2
import os

"""
给图片和对应标签重命名，适用于图片和标签一一对应的情况, opencv读取图像，不改变图像后缀
"""
img_path = './val/'
ann_path = './ann_txt/'
img_save_path = './images_drone/'
ann_save_path = './labels_drone_txt/'
if not os.path.exists(img_save_path):
    os.makedirs(img_save_path)
if not os.path.exists(ann_save_path):
    os.makedirs(ann_save_path)

imgs = os.listdir(img_path)
anns = os.listdir(ann_path)
# 保存图片原名前缀和改名后的前缀到字典中，用于对对应的标签改名，
# 适用于图片和标签一一对应的情况，就不需要每遍历一张图片而去遍历所有的标签,加快处理速度
old_new_name = {}
num_img =0
num_ann =0
for i ,img in enumerate(imgs):
    image = cv2.imread(img_path + img)
    img = img.split('.')
    name = img[0]
    new_name = 'drone_10_%d'%i
    old_new_name[name] = new_name
    img[0] = new_name
    img &