栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

跑通CHPDet模型

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

跑通CHPDet模型

一、运行环境
  • 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里面的路径,要正确。

三、运行代码 3.1 demo.py

可以下载官方的代码,确定程序可以跑通。直接运行 python demo.py multi_pose可能会出现下面错误。

找到网址下载好权值文件,保存到指定路径下就可以了。
还需要指定测试图像位置和权值文件路径。

至此,完成demo的操作,有时间将会更新训练和测试部分。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1025768.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号