mobilenet-yolov4-pytorch icon indicating copy to clipboard operation
mobilenet-yolov4-pytorch copied to clipboard

关于yolov4+mobilenetv3训练的问题

Open Yi19960820 opened this issue 4 years ago • 11 comments

您好,一开始不清楚您写了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%左右,是我的训练方法出了问题吗?

Yi19960820 avatar Dec 01 '20 13:12 Yi19960820

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

bubbliiiing avatar Dec 07 '20 07:12 bubbliiiing

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

miaowwy avatar Dec 29 '20 02:12 miaowwy

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

bubbliiiing avatar Dec 29 '20 08:12 bubbliiiing

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

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

miaowwy avatar Jan 04 '21 13:01 miaowwy

有FPS_test。

bubbliiiing avatar Jan 05 '21 04:01 bubbliiiing

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

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

ZZHHogan avatar Jan 11 '21 01:01 ZZHHogan

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

bubbliiiing avatar Jan 12 '21 14:01 bubbliiiing

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

wwwbobo avatar Nov 16 '21 08:11 wwwbobo

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

bubbliiiing avatar Nov 16 '21 14:11 bubbliiiing

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

bianluxuan avatar Apr 02 '22 11:04 bianluxuan

predict.py

bubbliiiing avatar Apr 02 '22 14:04 bubbliiiing