MMDetection Faster RCNN > Onnx Export

1. Pytorch 최신 버전으로 설치하려고 하니 내 PC에 설치된 cuda 10.2는 더이상 지원하지 않는다고 하여  11.3설치

2. ananconda 가상환경 생성 및 설치

> conda create -n mmdetection python=3.7
> conda activate mmdetection
> conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
> pip install onnx
> cd mmdetection
> python setup.py install
> pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html

3. 작업할 폴더로 이동한 뒤 mmdetection를 다운받는다

git clone https://github.com/open-mmlab/mmdetection

4. faster rcnn 사전 학습된 파일을 다음의경로에서 다운 받은 뒤 현재 작업 폴더로 복사한다

http://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth 

 

 

5. 다음의 명령어로 pth파일을 onnx파일로 변경한다

python tools/deployment/pytorch2onnx.py \
configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py \
../faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth \
--output-file ../faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.onnx \
--input-img demo/demo.jpg --test-img demo/demo.jpg  \
--shape 608 608 \
--show \
--verify \
--dynamic-export \
--cfg-options \
model.test_cfg.deploy_nms_pre=-1