mask rcnn에서 mmdetection의 배치사이즈 설정
292
9 asked
안녕하세요.
교수님 강의 잘듣고 있습니다.
Mask rcnn의 수업과정의
'Balloon 데이터를 이용한 Train 실습 - Train 실행 및 Gray Scale로 배경 적용한 Segmentation 구현하기'를 참고해서,
제가 가지고 있는 이미지 데이터에 학습을 시켜보고 있습니다.
MMDetection은 배치사이즈를 어디서 설정하는지 궁금합니다.
config 파일을 보면, 'mask_rcnn_r50_fpn.py'은 mmdetection/configs/base/models에 위치해 있는데,
배치사이즈로 볼만한 내용이,
다음 내용밖에는 없거든요.
cfg.auto_scale_lr = dict(enable=False, base_batch_size=62)
16을 62로 변경했는데, Epoch의 배치사이즈는 변하지 않고 그대로 10으로 설정되는것 같습니다.
Mask Rcnn은 배치사이즈를 어디서 설정하는지 궁금합니다.
Answer 2
1
안녕하십니까,
samples_per_gpu 설정을 변경하시면 됩니다. samples_per_gpu는 GPU 카드 한장당 batch_size를 의미합니다.
아래와 같이 Config 설정 시
cfg.data.samples_per_gpu = 4 이면 gpu 당 batch size를 4로 설정합니다.
감사합니다.
MMDetection 버전 이슈
0
30
2
강의 환경설정 질문
0
45
2
Custom Dataset에서의 polygon 정보 관련
0
92
3
cvat.ai 보안 수준이 궁금합니다
0
82
2
캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.
0
107
3
추론 결과의 Precision(또는 mAP) 평가 방법
0
89
2
mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.
0
62
2
runpod에서 google drive 연결 시 오류 발생
0
117
2
로드맵 선택
0
70
1
mmcv
0
62
2
Anchor box의 Positive 처리 위치
0
67
2
해당 강의 runpod 적용 후 에러 제보드립니다
0
90
2
run pod credit 관련 제보
0
109
2
mmdetection 2.x과 3.x 호환 관련 표기
0
81
2
mm_faster_rcnn_train_kitti.ipynb 실행 오류
0
108
3
질문 드립니다.
0
86
3
mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.
0
83
1
강사님께 수정을 제안드리고 싶은 것이 있습니다.
0
98
1
google automl efficientdet 다운로드 및 설치 오류
0
80
1
이상 탐지에 사용할 비전 기술 조언 부탁드립니다.
0
105
2
OpenCV 관련 질문드립니다.
0
79
2
mmcv 설치관련해서 문의드려요
0
340
3
강의 구성 관련해서 질문이 있습니다
1
139
2
모델 변환 성능 질문드립니다.
0
123
1