深度学习|r3det 配环境避雷指南(pytorch版)

1、创建新的conda环境
conda create -n r3det python=3.7
2、安装pytorch1.6.0、torchvision0.7.0、cuda10.1
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
3、卸载pillow,安装pillow6.2.2
pip uninstall pillow -y
pip install pillow==6.2.2
4、安装各种依赖库
pip install opencv-python==3.4.4.19
pip install mmcv==0.5.9
pip install pyparsing==2.3.0
pip install pycocotools==2.0.2
5、用预训练权重和图片做测试
python demo/image_demo.py demo/1.png configs/r3det/r3det_r50_fpn_2x_CustomizeImageSplit.py epoch_24.pth

要是想省事儿,直接下载下面的包,里面啥都有,不用额外下载权重和测试图了。
【深度学习|r3det 配环境避雷指南(pytorch版)】r3det压缩包链接深度学习|r3det 配环境避雷指南(pytorch版)
文章图片
https://download.csdn.net/download/wcybrain/44131349

    推荐阅读