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压缩包链接icon-default.png?t=LA92https://download.csdn.net/download/wcybrain/44131349

你可能感兴趣的:(深度学习,r3det,旋转目标检测,环境配置,pytorch)