我已经暂停训练自定义数据来训练我的自定义模型。但是 COCO 数据集中也有一些类/对象需要包含在内。有没有快速的方法从数据集中只提取那些类。我正在考虑编写一个脚本来做到这一点，但感觉应该有我更快的方法来实现这一点。
问问题
942 次
1 回答
0
您可以修改标签，也可以修改数据加载器以按类过滤。数据加载器在这里：
https://github.com/ultralytics/yolov3/blob/master/utils/datasets.py#L258
最后一个选项是对所有 80 个类进行训练，或者只使用预训练的权重，然后只对某些类运行检测：
python3 detect.py --classes 0,3,5
GitHub 上的 Glenn Jocher 回答了这个问题。原始问题链接
于 2020-04-06T18:09:42.967 回答