mobilenet-yolov4-pytorch
mobilenet-yolov4-pytorch copied to clipboard
关于yolov4+mobilenetv3训练的问题
您好,一开始不清楚您写了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%左右,是我的训练方法出了问题吗?
没啥问题,就是没用预训练权重的锅
您好,我正常训练,但是用video.py检测视频的时候fps才24左右,用的2080TI,请问是哪里出了问题呢
24正常,PIL绘图用了太多时间了
24正常,PIL绘图用了太多时间了
请问如果我想直接得到检测速度 应该怎么得到呢
有FPS_test。
没啥问题,就是没用预训练权重的锅
你好,请问“就是没用预训练权重的锅”是指loss有收敛,但是还是偏高这个问题吗?
没用coco的预训练权重,我上传了
@bubbliiiing up你好,这没看到coco预训练权重哎
这个库是VOC的预训练权重,上面那句说错了…库太多讲乱了
你好,大佬。请问FPS_test在您的库中的哪里呢?没有找到,感谢导师
predict.py