从coco数据集中提取子集-yolov5使用coco数据集子集训练测试
在yolov5源码中,找到data文件夹,新建一个yaml文件,文件名cat_dog.yaml,填入如下内容。这里选用yolov5x作为测试模型,需要找到yolov5–>models–>yolov5x.yaml文件。文件名 get_coco_datasets.py。本文使用coco2017,暂未验证其他数据集。将nc修改为你子集的class数量,这里为2。
原创
2023-10-19 11:22:52 ·
673 阅读 ·
5 评论