YOLO，提取与标签数量相等的图片，反之亦可

最新推荐文章于 2024-07-17 21:41:43 发布

下落的橘猫

最新推荐文章于 2024-07-17 21:41:43 发布

阅读量112

点赞数 1

文章标签： YOLO python

本文链接：https://blog.csdn.net/m0_66332641/article/details/131344066

版权

一、标签多图片少

file_root = r'D:\BaiduNetdiskDownload\CUHK\JPEG'  # 当前文件夹下的所有图片
file_list = os.listdir(file_root)

for i in file_list:
    if i.endswith('.jpg'):
        filename = r'D:\BaiduNetdiskDownload\CUHK\txt' + "\\" + i[:-4] + '.txt'   #所有标签
        print(os.path.exists(filename))
        if os.path.exists(filename):
            shutil.copy(filename, r'D:\BaiduNetdiskDownload\CUHK\JPEG')  #取出标签位置
            print(i + "处理成功！")

二、图片多标签少

file_root = r'D:\Dataset\sss'  # 当前文件夹下的所有标签
file_list = os.listdir(file_root)
print(file_list)

for i in file_list:
    if i.endswith('.txt'):
        filename = r'D:\BaiduNetdiskDownload\CityPersons\Citypersons\images\train' + "\\" + i[:-4] + '.jpg'   #所有图片
        print(os.path.exists(filename))
        if os.path.exists(filename):
            shutil.copy(filename, r'D:\BaiduNetdiskDownload\CityPersons\Citypersons\labels_with_ids\train2')  #取出图片位置
            print(i + "处理成功！")