我是靠谱客的博主 欣喜裙子,最近开发中收集的这篇文章主要介绍MobileNet-SSD,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

有预训练:

https://github.com/shaoshengsong/MobileNetV3-SSD

 

有v2 ssd:

https://github.com/qfgaohao/pytorch-ssd

 

MobileNetV3-SSD,在tx2上,没有明显优势。

https://github.com/tongyuhome/MobileNetV3-SSD

 

MobileNet-SSD 与yolov3对比,精度低了很多,具体就是漏检多了很多。

https://github.com/jacke121/Object-detection-state-of-the-art-RCNN-YOLO-SSD

 

用于训练的数据有30类,数据量大概在95k左右,trainval_proportion和train_proportion都是0.9。

第一部分的训练MobileNetV3模型,每个EPOCH的训练时间在10min左右(GTX1070),设置了20EPOCH之内没有收敛则停止训练,所以大概在140个EPOCH之后停止。训练之后在测试中能够达到99.60的准确率。

第二部分训练MobileNetV3 based SSD模型。这次的训练在进行了40个EPOCH停止,一方面是因为时间消耗太长,1070的算力不足,如果验证模型有效之后可以投入资源放置到服务器进行训练,所以用5类数据先进性训练会比较合理,另一方面为了得到更好的检测效果,希望尝试使用MobileNetV3-LARGE来代替现在使用的MobileNetV3-SMALL来进行训练。训练完成后我对这个MobileNetV3-SMALL based SSD - 40 EPOCH 模型进行了检测,结果如下:

最后

以上就是欣喜裙子为你收集整理的MobileNet-SSD的全部内容,希望文章能够帮你解决MobileNet-SSD所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(56)

评论列表共有 0 条评论

立即
投稿
返回
顶部