从coco数据库中选择特定的类别

• 图片和标签文件都在train2017中
• 新标签文件和对应的图片在另一个文件夹中，例子中是people_train

import os
from shutil import copyfile

src_path='train2017'
dst_path='people_train/'

labels = os.listdir(src_path)
labels.sort()

for label in labels:
  if label[-1] != 't':
    continue
  # 存标签
  tmp=[]
  for line in open(src_path+'/'+label):
    str_list=line.split()
    # 被选类别的标签
    if str_list[0]=='0':
      # 改成自己的标签
      line = '9'+line[1:]
      tmp.append(line)
  # 没有被选类别
  if len(tmp) < 1:
    continue
  # 新的标签文件
  with open(dst_path+label, 'w') as f:
   for item in tmp:
    f.write(item)
  #f.close()
  image=label[:-4]+'.jpg'
  # 拷贝有被选类别的图片
  copyfile(src_path+'/'+image,dst_path+image)