win10 centernet安装训练踩坑记录

最新推荐文章于 2022-02-23 16:32:47 发布

cffer123

最新推荐文章于 2022-02-23 16:32:47 发布

阅读量640

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/cffer123/article/details/115412375

版权

配置环境

前面的创建环境安装包略过。
我的cuda版本是10.2，pytorch版本是1.5.0，torchvision版本是0.6.0，python版本是3.7。（有人说在windows上torch版本只能是1.2.0，但是我的1.5.0也配置成功了。）安装完torch和torchvision后，安装requriments.txt 要求的包。 假设你上面全部顺利。
重头戏：编译DCNv2，从这里下载DCNv2替换到centernet自带的注意这里一定要用VS2017或者VS2015工具自带的命令行，如下图所示。

使用倒数第二个VS 2017x64本机工具。（如果你使用的是VS2015，同理，但是VS2015比2017多几个命令行工具，别找错了就行）。进入命令行工具，激活你刚刚创建的环境，然后进入到你下载的DCNv2里面，更改下面内容，然后进行编译。
首先定位到拟新建环境anaconda下面的目录：

修改两行代码：

comma

关注