Ubuntu上安装运行 rotated_maskrcnn

原创已于 2022-11-23 14:51:39 修改 · 660 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#深度学习

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

pytorch 专栏收录该内容

6 篇文章

订阅专栏

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

版本使用的是 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_minival2014.tar.gz
mkdir -p datasets/coco
ln -s /path_to_coco_dataset/annotations datasets/coco/annotations
ln -s /path_to_coco_dataset/train2014 datasets/coco/train2014
ln -s /path_to_coco_dataset/test2014 datasets/coco/test2014
ln -s /path_to_coco_dataset/val2014 datasets/coco/val2014

change the torch.uint8 to torch.bool:
maskrcnn_benchmark/structures/segmentation_mask.py
maskrcnn_benchmark/modeling/balanced_positive_negative_sampler.py

python tools/train_net.py --config-file “configs/rotated/e2e_ms_rcnn_R_50_FPN_1x.yaml” SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025 SOLVER.MAX_ITER 720000 SOLVER.STEPS “(480000, 640000)” TEST.IMS_PER_BATCH 1

train your own dataset:
labelme https://github.com/wkentaro/labelme/tree/main/examples/instance_segmentation
./labelme2coco.py data_annotated data_dataset_coco --labels labels.txt

Convert your own dataset to COCO annotation format (a json file). i use the labelme to do it.
Add the path of the dataset image directory and coco-format annotation file (from Step 1) into maskrcnn_benchmark/config/paths_catalog.py remember change the num_class
In your .yaml config file (e.g. “configs/rotated/e2e_ms_rcnn_R_50_FPN_1x.yaml”), change the DATASETS.TRAIN value to the stuff you added in paths_catalog.py.
change the MODEL.ROI_HEADS.NMS to 0.3