Darknet训练yolov7-tiny(AlexeyAB版本)

时间:2025-03-13 07:32:24
根据数据集进行相关参数修改,如果使用上述coco2017数据集,那么数据类别为80类,直接使用cfg进行训练即可。
如果你没有使用coco数据集,而使用自己的数据进行训练,那么需要修改配置cfg文件:
修改classes,使用快捷键搜索关键字[yolo]可以搜到2次,修改classes的数量为你的类别数,这里classes=2,
修改filters,每次搜到的yolo上一个的[convolutional]中filters=(classes + 5)x3
比如filters=21。