根据Main/train_all.txt中的文件名，从labels_all中挑选出训练集的标签文件

原创于 2022-09-02 10:35:12 发布 · 147 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #人工智能

yolov3 专栏收录该内容

15 篇文章

订阅专栏

# 根据Main/train_all.txt中的文件名，从labels_all中挑选出训练集的标签文件

import shutil, glob, os

sets = ['train_all',  'test_all']
label_ids = os.listdir('/Users/h1y/Documents/mac2server/echi/echi/labels_all')

old_label_path = '/Users/h1y/Documents/mac2server/echi/echi/labels_all'
new_label_path = '/Users/h1y/Documents/mac2server/echi/echi/labels_train'

image_set = 'train_all',
    # 0001
image_ids = open('/Users/h1y/Documents/mac2server/echi/echi/ImageSets/Main/%s.txt' % (image_set)).read().strip().split()

for label_id in label_ids:
    if label_id[:-4] in image_ids:
        shutil.copyfile(old_label_path + '/'+label_id, new_label_path+'/'+label_id)