search

bubbliiiing / mobilenet-yolov4-pytorch

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

关于mobilenetv3预训练权重 #11

Open mei123hao opened 3 years ago

mei123hao commented 3 years ago

大神你好，感谢你提供的代码。我也尝试用mobilenetv3主干网络去修改yolo，用了这个仓库 https://github.com/d-li14/mobilenetv3.pytorch.git 所提供的权重。但是训练测试结果貌似没有达到你的指标。只有48%的map0.5。而且训练过程发现loss没办法收敛到1一下。但是我用darknet53的主干网是可以的，请问是不是它所提供的预训练权重的问题？您训练出来的voc权重也是基于这个仓库所提供的主干网络权重去训练的吗？ image

bubbliiiing commented 3 years ago

我不是提供了预训练权重吗，mobilenetv3光主干的

mei123hao commented 3 years ago

对，我就想知道您所提供的主干是不是来自那个仓库的？如果是的话那可能就不是主干的原因了，是我其他地方的问题

bubbliiiing commented 3 years ago

我看了一下，和我权值名字一样，应该是一样的