发布于2019-08-20 10:23 阅读(5402) 评论(0) 点赞(3) 收藏(3)
PyTorch实现 MobileNetV3-SSD 用于目标检测
flyfish
MobileNetV3-SSD implementation in PyTorch
环境
操作系统: Ubuntu18.04
Python: 3.6
PyTorch: 1.1.0
下载地址:
https://github.com/shaoshengsong/MobileNetV3-SSD
使用MobileNetV3-SSD实现目标检测
代码参考(严重参考以下代码)
一 SSD部分
https://github.com/amdegroot/ssd.pytorch
A PyTorch Implementation of Single Shot MultiBox Detector
二 MobileNetV3 部分
1 https://github.com/xiaolai-sqlai/mobilenetv3
mobilenetv3 with pytorch,provide pre-train model
2 https://github.com/kuan-wang/pytorch-mobilenet-v3
MobileNetV3 in pytorch and ImageNet pretrained models
3 https://github.com/leaderj1001/MobileNetV3-Pytorch
Implementing Searching for MobileNetV3 paper using Pytorch
4 https://github.com/qfgaohao/pytorch-ssd (严重抄袭此处)
MobileNetV1-SSD/SSD-lite,MobileNetV2-SSD/SSD-lite部分
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
no discernible latency cost.
针对4我这里没有做MobileNetV1, MobileNetV2等等代码兼容,只有MobileNetV3可用
下载数据
本例是以蛋糕和面包为例,原因是数据量小
所有类别总大小是561G,蛋糕和面包是3.2G
python3 open_images_downloader.py --root /media/santiago/a/data/open_images --class_names “Cake,Bread” --num_workers 20
训练过程
首次训练
python3 train_ssd.py --dataset_type open_images --datasets /media/santiago/data/open_images --net mb3-ssd-lite --scheduler cosine --lr 0.01 --t_max 100 --validation_epochs 5 --num_epochs 100 --base_net_lr 0.001 --batch_size 5
预加载之前训练的模型
python3 train_ssd.py --dataset_type open_images --datasets /media/santiago/data/open_images --net mb3-ssd-lite --pretrained_ssd models/mb3-ssd-lite-Epoch-99-Loss-2.5194434596402613.pth --scheduler cosine --lr 0.01 --t_max 100 --validation_epochs 5 --num_epochs 200 --base_net_lr 0.001 --batch_size 5
测试一张图片
python run_ssd_example.py mb3-ssd-lite models/mb3-ssd-lite-Epoch-99-Loss-2.5194434596402613.pth models/open-images-model-labels.txt /home/santiago/picture/test.jpg
作者:comeonbady
链接:https://www.pythonheidong.com/blog/article/48971/e61a1bf42aeb85e012cd/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!