SSD-pytorch训练自己的数据(附数据)

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

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

评论 25
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值