Windows下自己训练数据集,在pytorch上实现基于SSD的物体检测
(0)环境
Windows10
Anaconda3(python3.7.6)
GPU:GTX1660Ti-Max-Q
cuda10.0
cudnn7.6.5
pytorch1.2
(一)准备数据集
我用的labelImg标注,图片来源百度搜索图片:狗(共40张)。具体标注方法网上很多,不赘述。
数据集链接:链接:https://pan.baidu.com/s/1We-XseyDVmSa-JWN1om0QQ 提取码:afqk
(二)下载源码并修改
SSD-Pytorch代码链接:https://github.com/amdegroot/ssd.pytorch
1、data/__init__.py 第3行注释from .coco import COCODetection, COCOAnnotationTransform, COCO_CLA

本文介绍了在Windows10上使用PyTorch实现基于SSD的物体检测,包括环境配置、数据集准备、源码下载与修改、训练过程以及测试效果。作者使用40张狗的图片进行标注,并分享了训练过程中的参数调整和遇到的问题,如loss变为nan的解决方法。最后,提供了训练代码和数据集的下载链接。
1万+

被折叠的 条评论
为什么被折叠?



