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 进行安装即可