Ubuntu上安装运行 rotated_maskrcnn

doublechenchenchen

已于 2022-11-23 14:51:39 修改

阅读量594

点赞数

分类专栏： pytorch 文章标签：深度学习

于 2022-07-08 09:35:14 首次发布

本文链接：https://blog.csdn.net/doublechenchenchen/article/details/125671929

版权

本文详细介绍了在Ubuntu系统中，如何配置和安装环境以运行rotated_maskrcnn，包括依赖库的安装、COCO API的获取、apex的安装以及训练自己的数据集等步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

版本使用的是 torch1.4.0
cuda 使用的是cuda10.0

export CUDA_HOME=/usr/local/cuda
fist:
install dependency:
conda create --name rotated_maskrcnn python=3.6
conda activate rotated_maskrcnn
conda install ipython
pip install ninja yacs cython matplotlib tqdm opencv-python
（注意 opencv-python 安装4.4.0.*版本之前用4.3版本会出问题哦。那个cv2,minirect那会有问题）
conda install pytorch1.4.0 torchvision0.5.0 cudatoolkit=10.0 -c pytorch

git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install

git reset --hard 3fe10b5597ba14a748ebb271a6ab97c09c5701ac
git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext

git clone https://github.com/mrlooi/rotated_maskrcnn.git
cd rotated_maskrcnn
python setup.py build develop

http://datasets.d2.mpi-inf.mpg.de/hosang17cvpr/coco_miniv