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

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

3019f38cdaa549f07ee8f0b1e1e372a7.png

pycharm建立新python环境

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

0f8524c249350f67476d7998122eb198.png

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

63b7d6b070d6ee71afd5f1a880695596.png

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进行离线安装,如下图所示

e1d9191f37c15e6ea4c8dae6c75381a5.png

通过pycharm来安装pytorch库

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

7a66e0541e60d4e59e3cadef3764353d.png

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

561fae95852aa239ee9c5efa4669af54.png

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

dee550d39028f061dc76d1e27204fc9c.png

再输入下图红框参数

a29524dc65f909f9028a7eb2edc7dcf6.png

安装成功

2c247bb071699c301cbb1abdb8548eec.png

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

0e403c4e1149ca3de0acc0ff856638ed.png

7.准备自己的数据集

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐