- python 3.9.12
- torch 1.12.0+cu116
- numpy==1.22.4
- pycocotools==2.0
- Cython==0.29.30
项目github地址: https://github.com/zf020114/CHPDet
需要安装的有:
- lib/models/networks/DCNv2
- lib/models/networks/orn
- lib/models/external
如果直接运行python setup.py build develop,因为pytorch版本问题,会出现:
前两项安装在目前的环境下是无法成功的,我们需要修改一些代码。主要原因是新版的pytorch删去了THC/THC.h函数。
对于DCNv2网上有现成的新版代码适合新版pytorch。我们下载下来替换即可。网址:https://gitcode.net/mirrors/jinfagang/DCNv2_latest
对于orn目前网上没有找到新的代码,我们只能自己修改。参考:https://github.com/CoinCheung/pytorch-loss/pull/37
https://blog.csdn.net/qq_36891089/article/details/124353149
template__host__ __device__ __forceinline__ T THCCeilDiv(T a, T b) { return (a + b - 1) / b; } template __host__ __device__ __forceinline__ T THCRoundUp(T a, T b) { return THCCeilDiv(a, b) * b; }
对于第三项的安装,可以直接运行python setup.py build develop
前面的安装过程如果出现 /orn/csrc/vision.cpp' cannot be absolute,这时可以看一下egg-info/SOURCES.txt里面的路径,要正确。
可以下载官方的代码,确定程序可以跑通。直接运行 python demo.py multi_pose可能会出现下面错误。
找到网址下载好权值文件,保存到指定路径下就可以了。
还需要指定测试图像位置和权值文件路径。
至此,完成demo的操作,有时间将会更新训练和测试部分。



