2021.10.19 - [딥러닝관련/Detection] - MMdetection(1) 설치 및 Demo 실행 - (1)
MMdetection(1) 설치 및 Demo 실행 - (1)
MMdetection MMdetection은 openMMlab이라고 The CUHK Multimedia Lab에서 제공하는 Open Source Project의 프로젝트 중 하나이다. (https://mmlab.ie.cuhk.edu.hk/) 아래 사이트에 보면 MMDetection 말고도 다른..
better-tomorrow.tistory.com
위 환경 설정 및 데모에 이어 COCO, Pascal VOC 등 학습을 진행해보고자 한다.
데이터 구조는 아래를 참고하여 다운 받아 구성한다.
https://detectron2.readthedocs.io/en/latest/tutorials/builtin_datasets.html
Use Builtin Datasets — detectron2 0.6 documentation
Use Builtin Datasets A dataset can be used by accessing DatasetCatalog for its data, or MetadataCatalog for its metadata (class names, etc). This document explains how to setup the builtin datasets so they can be used by the above APIs. Use Custom Datasets
detectron2.readthedocs.io
학습 방법은 아래를 참고한다.
Getting Started with Detectron2 — detectron2 0.6 documentation
This document provides a brief intro of the usage of builtin command-line tools in detectron2. For a tutorial that involves actual coding with the API, see our Colab Notebook which covers how to run inference with an existing model, and how to train a buil
detectron2.readthedocs.io
아래 명령어를 사용해 데이터셋의 위치를 고정시킨 후
export DETECTRON2_DATASETS=/path/to/datasets학습 실행
./train_net.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml --num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025그러면 아래와 같이 학습이 진행되는 것을 알 수 있다.
'딥러닝관련 > Detection' 카테고리의 다른 글
| 병변 검출 AI 검진대회 기록 (비공개) (0) | 2021.11.29 |
|---|---|
| Object detection 정리 (1) (feat, object detection? , 1 stage detector, 2 stage detector) (0) | 2021.11.03 |
| MMdetection(2) - 표준 Datasets Inference 및 Train (0) | 2021.10.28 |
| MMdetection(1) 설치 및 Demo 실행 - (2) (config) (0) | 2021.10.27 |
| MMCV 의 Registry (0) | 2021.10.26 |
