COCO数据集提取特定多个类并在YOLO-V3上训练

先占个地方,有空再写
`
import os

Dir = './coco_class_6/Annotations/val2014'
ImageDir = './coco_class_6/images/val2014'
cnt = 0
for i, file_name in enumerate(os.listdir(Dir)):
fsize = os.path.getsize(os.path.join(Dir,file_name))
if fsize == 410:
print('removing {} of size{}'.format(file_name,fsize))
os.remove(os.path.join(ImageDir, file_name[:-3]+'jpg'))
os.remove(os.path.join(Dir, file_name))
cnt += 1

print('remove {} files'.format(cnt))
`

猜你喜欢

转载自www.cnblogs.com/zyxxmu/p/12935105.html