判断图片（images）和标签（labels）是否对应，并同时更改文件名

Kill_Spring

于 2020-10-02 00:32:48 发布

阅读量2.9k

点赞数 5

CC 4.0 BY-SA版权

文章标签：深度学习机器学习 os

本文链接：https://blog.youkuaiyun.com/Kill_Spring/article/details/108898541

import os
train_images_path_old = r'C:\Users\Administrator\Desktop\mask\train\images'
train_images_path_new = r'C:\Users\Administrator\Desktop\mask\train\images2'
train_labels_path_old = r'C:\Users\Administrator\Desktop\mask\train\labels'
train_labels_path_new = r'C:\Users\Administrator\Desktop\mask\train\labels2'

os.chdir(train_images_path_old)
os.chdir(train_labels_path_old)
list_images = os.listdir(train_images_path_old)
list_labels = os.listdir(train_labels_path_old)
images_number = 0 #图片后面加的序号
labels_number = 0 #图片后面加的序号,用一个就行
for images_list_num in range(0, len(list_images)):  #先遍历图片序号，再检测标签需要是否与之对应，之后改成一样的名字

    for labels_list_num in range(0, len(list_labels)):
       images_old_name = list_images[images_list_num]
       labels_old_name = list_labels[labels_list_num]
       if (images_old_name[:-3]==labels_old_name[:-3]):
            print("图片文件与标签已经对应，对应的序号为&#x