训练自己的数据集
感谢 大佬的开源!!!
DataXujing
我们以训练YOLOv4-P7为例,介绍如何基于Scaled YOLOv4训练自己的数据集
0.环境配置
python3.7 cuda 10.2
pytorch==1.6.0
torchvision==0.7.0
# mish-cuda
# 使用预训练的模型
git clone https://github.com/thomasbrandon/mish-cuda mc
cd mc
# change all of name which is mish_cuda to mish_mish and build.
# 1. mc/src/mish_cuda -> mc/src/mish_mish
# 2. mc/csrc/mish_cuda.cpp -> mc/csrc/mish_mish.cpp
# 3. in mc/setup.p
1