yolov5如何运行自己的数据集_win10系统下centernet(2019cvpr)训练自己的数据集
1.从github下载centernet https://github.com/xingyizhou/CenterNet2.通过pycharm来建立centernet所需要的环境,因为每个模型所需要的库可能不一样,当然也能通过cmd命令行来进行安装库。点击pycharm界面右上角file,点击setting,如下图,左边找到project ->project interpreter->
1.从github下载centernet https://github.com/xingyizhou/CenterNet
2.通过pycharm来建立centernet所需要的环境,因为每个模型所需要的库可能不一样,当然也能通过cmd命令行来进行安装库。点击pycharm界面右上角file,点击setting,如下图,左边找到project ->project interpreter->creat virtualenv 来建立新的环境,我所建立的环境为centernet-pytorch。

pycharm建立新python环境
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进行离线安装,如下图所示

通过pycharm来安装pytorch库
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.准备自己的数据集

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