공부 일기장

mmdetection 사용 2 - train 본문

네이버 부스트 캠프/mmdetection 사용법

mmdetection 사용 2 - train

곰후니

train하기 앞서서 _base_와 비슷한 형태로 세팅을 해두고

아래와 같이 test_model.py를 만들어두자

_base_ = [
    './_base_/datasets/coco_detection.py',
    './_base_/models/faster_rcnn_r50_fpn.py',
    './_base_/schedules/schedule_1x.py',
    './_base_/default_runtime.py'
]

 

그리고 터미널에 python train.py configs/test_model.py 해주면 학습이 시작된다.

train.py 하는 경우 설정해줄수 있는 config가 여러가지 seed고정을 하는건 필수적이라고 본다.

Comments