从coco数据集中提取需要的类别

进行目标检测时，有时只需要训练数据集中的部分图像，以 coco128 为例，只选出其中的车辆类：bicycle car motorcycle bus truck。
coco128 数据集中的标签为 txt 文件，每一个图像由若干行，每一行对应一个目标的类别序号和 4 个坐标（中心 x，中心 y，宽，高，只需要选出指定类别序号的 txt 文件，然后保存同名的图像文件即可。coco 数据集同理。

# 从coco128中提取车类的image和label
# 0 bicycle
# 1 car
# 2 motorcycle
# 3 bus
# 4 truck

import os
from shutil import copyfile

src_path = './coco128/labels/train2017/'  # 标签
img_path = './coco128/images/train2017/'  # 图像
dst_label_path = './traffic_train_coco128/labels/'
dst_img_path = './traffic_train_coco128/ima