search

bubbliiiing / mobilenet-yolov4-pytorch

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

关于yolov4+mobilenetv3训练的问题 #1

Open Yi19960820 opened 3 years ago

Yi19960820 commented 3 years ago

您好，一开始不清楚您写了mobilenet-yolov4-lite-pytorch的代码，我将您的yolov4 pytorch版本的cspdarknet53替换成了mobilenetv3，操作的一样的，唯一不同的是我还使用了v3里面的最后一层卷积层，输出维度为13x13x960，并改动了yolo4.py里面三次卷积，五次卷积以及上下采样的维度，我用3080训练了400个epoch： lr = 1e-3 Batch_size = 8 Init_Epoch = 0 Freeze_Epoch =200 lr = 1e-5 Batch_size = 8 Freeze_Epoch = 50 Unfreeze_Epoch = 200 无预训练权重，voc2012数据集，loss在收敛于26.3，map为34.5％，速度比yolov4快20％左右，是我的训练方法出了问题吗？

bubbliiiing commented 3 years ago

没啥问题，就是没用预训练权重的锅

miaowwy commented 3 years ago

您好，我正常训练，但是用video.py检测视频的时候fps才24左右，用的2080TI，请问是哪里出了问题呢

bubbliiiing commented 3 years ago

24正常，PIL绘图用了太多时间了

miaowwy commented 3 years ago

24正常，PIL绘图用了太多时间了

请问如果我想直接得到检测速度 应该怎么得到呢

bubbliiiing commented 3 years ago

有FPS_test。

ZZHHogan commented 3 years ago

没啥问题，就是没用预训练权重的锅

你好，请问“就是没用预训练权重的锅”是指loss有收敛，但是还是偏高这个问题吗？

bubbliiiing commented 3 years ago

没用coco的预训练权重，我上传了

wwwbobo commented 2 years ago

@bubbliiiing up你好，这没看到coco预训练权重哎

bubbliiiing commented 2 years ago

这个库是VOC的预训练权重，上面那句说错了…库太多讲乱了

bianluxuan commented 2 years ago

你好，大佬。请问FPS_test在您的库中的哪里呢？没有找到，感谢导师

bubbliiiing commented 2 years ago

predict.py