0. 杂言
最近抱着学习的心态开启了新一轮的学习任务,耗时4天,圆满运行成功。
由于github多数情况下并没有办法链接,建议使用gitee或gitcode 等国内的管理平台直接下载
1.下载文件
本文的多数文件是可以直接下载的附上链接如下
centernet2的下载:centernet , centernet
在读完centernet2中的README.md,对大体的情况有了一定的了解,我只能说写的真的详细(🔨),不如不写。
了解到安装detectron2是复现centernet2的前提后,那么就不要多想,下。
detectron2的下载:detectron2 这个属于源码能看但跑不了,win上我没跑处理 跑出来的detectron2,这个应该是下不了的,需要的私我。自己有渠道的自己下,不大。
这两个的是detectron2对应需要下载的fvcore,cocoapi 。(注:下到detectron2内)
具体的介绍文档首先可以看下官方文档,或者比较好点的博客,这个博客算的上一份较好的说明文档可以在下载部分模型的无聊时间看看,有个大致的了解为后期做一点准备。
2 安装环境与成功运行的log文本
conda create -n detectron2 python==3.6
conda activate detectron2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
创建好环境后对fvcore和cocoapi进行编译
# cd 到对应的文件夹cocoapi/PythonAPI内
python setup.py build_ext install
# cd 到 vcore 内
python setup.py build --force develop
对detectron2中的进行构建cd detectron2中
python setup.py build --force develop
当然是跑不了的啦,改。
3.修改文件
接下就是对着改了
第一个是文本问题
问题1修改torch\utils\cpp_extension.py
位置是342行改为decode(‘gbk’),(注意:框的那个是linux的不要管)
问题2 具体修改的地方见他的问题2.1和2.2
最后就是
python setup.py build --force develop
到此你不会就以为可以demo了吧就应该会遇到一个版本问题future feature annotations is not defined(不要搜索了,网上许多瞎扯答案)
卸载当前版本的pip uninstall termcolor
安装 pip install termcolor 大概是1.1.0的
其余的都是未安装的包的问题,如果不会,建议重新学。
4.完成
最后demo.py
参考第一张图
# path/to/image/这个写你自己的文件夹,不要懒
python demo.py --config-file configs/CenterNet2_R50_1x.yaml --input path/to/image/ --opts MODEL.WEIGHTS models/CenterNet2_R50_1x.pth
系统是linux,detectron2安装更友好
centernet2的训练部分,再说吧。有1000人看就发,没得就算了。再见!!!