当前位置: 首页 > 工具软件 > RPNet > 使用案例 >

CCPD - rpnet 代码复盘

齐冥夜
2023-12-01



CCPD - rpnet 网络复盘

rpnet的代码库:https://github.com/detectRecog/CCPD

一、环境准备

1、环境版本依赖

  • python: pytorch(0.3.1), numpy(1.14.3), cv2(2.4.9.1).
  • system: Cuda(release 9.1, V9.1.85)

2、环境搭建

2.1、Cuda 部分:

参考:https://blog.csdn.net/pentiumCM/article/details/108856508

2.2、python:

  • 使用 anaconda 创建虚拟环境:
conda create -n rpnet python=3.6

  • 安装 pytorch:
    (这部分需要格外注意,pytorch版本和作者不一致,运行代码会有很多问题,尽可能保持一致)
pip3 install http://download.pytorch.org/whl/cu91/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision==0.2.1

参考:https://ptorch.com/news/145.html
–(使用conda/pip安装pytorch 0.3.1教程(mac/windos/linux))

  • 其他软件依赖正常通过 pip / conda 进行安装即可
 类似资料: