欢迎大家来到IT世界,在知识的湖畔探索吧!
1.从github下载centernet https://github.com/xingyizhou/CenterNet
2.通过pycharm来建立centernet所需要的环境,因为每个模型所需要的库可能不一样,当然也能通过cmd命令行来进行安装库。点击pycharm界面右上角file,点击setting,如下图,左边找到project ->project interpreter->creat virtualenv 来建立新的环境,我所建立的环境为centernet-pytorch。
3.通过pycharm打开工程,如下图会出现好多的模块无法导入的情况
因为该模型的源码是在liunx系统下运行的,所以可能会出现模块导入的问题,接下来修改导入库问题(参考文献https://blog.csdn.net/u013210620/article/details/78613886),修改如下图所示
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进行离线安装,如下图所示
5.点mian函数进行调试,报错torchvision模块没有
那就下载相应的包,参考文献(https://blog.csdn.net/miao0967020148/article/details/80400536)我下载的版本是0.1.6,试了高版本不行
6.运行接着报错,没有pycocotools库,安装它,但pycocotools大佬实在linux系统下写的,需要下载其它大佬改写的win10版本,下载地址(https://github.com/philferriere/cocoapi),然后就是通过pycharm进行安装,打开工程cocoapi-master运行setup.py会显示没有参数输入,那就输入,先输入图中红框中参数
再输入下图红框参数
安装成功
接下来运行没有错误了,就是输入模型的一些参数
7.准备自己的数据集
免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://itzsg.com/34984.html