
Mobilenet-YOLO-Pytorch:涵盖mobilenet系列(v1,v2,v3...)及yolo系列(yolov3...)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Mobilenet-YOLO-Pytorch是一个基于PyTorch框架开发的项目,集成了多种高效的轻量级模型和目标检测算法,包括Mobilenet v1/v2/v3及Yolov3等。
Mobilenet-YOLO-Pytorch 与我之前的项目相似,其损失函数与原始实现非常接近。该模型使用PyTorch实现了基于MobileNet的YOLO检测网络,并在VOC2007(包含07+12)数据集上进行了训练和测试。预训练图像网络未采用Coco。
以下是不同配置的结果:
- MobileNetV2:352分辨率,精度为71.2
- MobileNetV3
要开始使用这个项目,请按照以下步骤操作:
1. 下载并解压VOCdevkit数据集(如果已存在该数据集,则可以跳过此步)。
2. 运行脚本以创建lmdb文件:`sh scripts/create.sh`
3. 开始训练模型:`sh scripts/train.sh`
演示版本可以通过执行相应的下载命令来获取,并将其保存在$ Mobilenet-YOLO-Pytorch / check目录中。
全部评论 (0)
还没有任何评论哟~


