bubbliiiing / mobilenet-yolov4-pytorch

这是一个mobilenet-yolov4的库,把yolov4主干网络修改成了mobilenet,修改了Panet的卷积组成,使参数量大幅度缩小。
MIT License
369 stars 104 forks source link

训练VOC数据集时的参数以及电脑配置 #49

Open Jie-Huangi opened 1 year ago

Jie-Huangi commented 1 year ago

B导,我在你mobilenet-yolov4-pytorch的基础上换了一个ghostnetv2网络。现在我想和你之前的主干ghostnetv1、mobilenetv1-3等网络对比一下性能,但是不知道您之前使用VOC数据集训练网络时给的参数配置,想麻烦B导给个epoch,batch-size,optimizer,等其它参数呢。

bubbliiiing commented 1 year ago

之前?

bubbliiiing commented 1 year ago

https://github.com/bubbliiiing/mobilenet-yolov4-pytorch/blob/bilibili/train.py

Jie-Huangi commented 1 year ago

嗯嗯,B导,我将ghostnetv2网络使用您给的voc数据训练了300个epoch,map0.5最好的一次才67.20%。远远低于其它主干网络的准确性。难受了,是我训练有问题还是网络不行哦,我是使用在线服务器训练的,300个epoch得用1天半30块钱左右。