win10系统下centernet(2019cvpr)训练自己的数据集

win10系统下centernet(2019cvpr)训练自己的数据集win10系统下centernet(2019cvpr)训练自己的数据集

欢迎大家来到IT世界,在知识的湖畔探索吧!

1.从github下载centernet https://github.com/xingyizhou/CenterNet

2.通过pycharm来建立centernet所需要的环境,因为每个模型所需要的库可能不一样,当然也能通过cmd命令行来进行安装库。点击pycharm界面右上角file,点击setting,如下图,左边找到project ->project interpreter->creat virtualenv 来建立新的环境,我所建立的环境为centernet-pytorch。

win10系统下centernet(2019cvpr)训练自己的数据集

pycharm建立新python环境

3.通过pycharm打开工程,如下图会出现好多的模块无法导入的情况

win10系统下centernet(2019cvpr)训练自己的数据集

因为该模型的源码是在liunx系统下运行的,所以可能会出现模块导入的问题,接下来修改导入库问题(参考文献https://blog.csdn.net/u013210620/article/details/78613886),修改如下图所示

win10系统下centernet(2019cvpr)训练自己的数据集

4.如果torch无法导入是因为没有安装pytorch,接下来通过pycharm来安装。pytorch下载(https://pytorch.org/get-started/previous-versions/)我的环境是win10,python3.6,cuda9.0下载的版本为 cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl,之所以为0.4.1版本,是因为模型用的是0.4.1。下载好之后,点pycharm终端采用pip install进行离线安装,如下图所示

win10系统下centernet(2019cvpr)训练自己的数据集

通过pycharm来安装pytorch库

5.点mian函数进行调试,报错torchvision模块没有

win10系统下centernet(2019cvpr)训练自己的数据集

那就下载相应的包,参考文献(https://blog.csdn.net/miao0967020148/article/details/80400536)我下载的版本是0.1.6,试了高版本不行

win10系统下centernet(2019cvpr)训练自己的数据集

6.运行接着报错,没有pycocotools库,安装它,但pycocotools大佬实在linux系统下写的,需要下载其它大佬改写的win10版本,下载地址(https://github.com/philferriere/cocoapi),然后就是通过pycharm进行安装,打开工程cocoapi-master运行setup.py会显示没有参数输入,那就输入,先输入图中红框中参数

win10系统下centernet(2019cvpr)训练自己的数据集

再输入下图红框参数

win10系统下centernet(2019cvpr)训练自己的数据集

安装成功

win10系统下centernet(2019cvpr)训练自己的数据集

接下来运行没有错误了,就是输入模型的一些参数

win10系统下centernet(2019cvpr)训练自己的数据集

7.准备自己的数据

免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://itzsg.com/34984.html

(0)

相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

联系我们YX

mu99908888

在线咨询: 微信交谈

邮件:itzsgw@126.com

工作时间:时刻准备着!

关注微信