如何从YOLOv3的COCO数据集中删除一些对象

Question

我已经休假Train Custom Data来训练我的自定义模型。但是，COCO数据集中有一些类/对象也要包含在内。有没有一种快速方法可以从数据集中仅提取那些类。我正在考虑编写一个脚本来做到这一点，但感觉应该有我要快得多的方法来实现。

Answer 1

您可以修改标签，也可以修改数据加载器以按类别进行过滤。数据加载器在这里：

https://github.com/ultralytics/yolov3/blob/master/utils/datasets.py#L258

最后一个选择是对所有80个班级进行训练，或者仅使用预训练的权重，然后仅对某些班级进行检测：

python3 detect.py-类0,3,5

Glenn Jocher在GitHub上回答了这个问题。原始问题link