文章目录
整体代码结构：
一、cfg文件修改
创建yolov4-mydatasets.cfg，因为我用了yolov4.cfg训练太吃显存，2080TI的机子batchsize设置为2也会爆显存。其主要原因是mish函数太占显存了。
修改的地方：
将mish函数替代为relu函数，大大降低显存使用！
另外filters设置为：
（class数目 + 4 +1） × 3 = 45
（我这里10个类别）分别对应三个detect header
classes=10 修改为类别数目
版权声明：本文为wa1tzy原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。