关于在windows上完成目标检测模型Centernet 以及自己数据的训练
关于在windows上完成目标检测模型Centernet 以及自己数据的训练
配置设置
最近在windows上复现了centernet模型,跑通了demo,并对自己的数据进行了训练,这里对过程进行记录。
配置:Windows 10,VS2015,Anaconda3.5.1,python3.6,torch1.1,Cuda10.0,Cudnn7.5。(VS2015和python3.6应该是对应torc...
原创
2020-01-15 20:51:42 ·
2405 阅读 ·
22 评论