mmdetection에서 custom-dataset 학습하기
423
작성한 질문수 1
coco-dataset형태로 json형태의 train.json 형태로 custom-dataset을 생성하여 pretrained된 모델에 학습하는 과정에서, Custom-dataset의 CLASSES = ('WBC', 'RBC', 'Platelets')로 classes를 정의해주는 것은 이해했습니다.
그러면 모델 자체가 coco-data로 pretrained된 모델인 것으로 알고 있는데, CLASSES를 3개로 정하면, coco dataset으로 훈련된 나머지 80개의 object는 더 이상 detect하지 않는건가요?
아니면, 80개에 제가 추가해준 3개를 추가로 detect할수 있는 건가요?
답변 1
0
안녕하십니까,
COCO로 Pretrained 된 모델의 weight만 가져 오는 것입니다. 실제 학습 데이터는 3개의 클래스를 가지는 blood cell 데이터 입니다. 학습하는 모델은 3개의 클래스를 가지는 모델이므로 3개의 클래스만 Detect가 가능합니다. 다만 학습을 수행하기 이전에 coco로 pretrained된 모델의 weight를 그대로 3개의 클래스를 가지는 모델의 weight로 전이하는 것입니다. 최종 classification을 하는 layer는 이때 가져오지 않습니다.
요약드리면 3개의 클래스를 가지는 classification layer를 모델로 적용하되, 이 classification layer를 제외한 다른 layer들의 weight값을 coco로 pretrained된 모델의 weight값으로 대체하는 것입니다. 때문에 3개의 class만 Detect가 가능합니다.
MMDetection 버전 이슈
0
53
2
강의 환경설정 질문
0
62
2
Custom Dataset에서의 polygon 정보 관련
0
113
3
cvat.ai 보안 수준이 궁금합니다
0
98
2
캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.
0
120
3
추론 결과의 Precision(또는 mAP) 평가 방법
0
96
2
mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.
0
69
2
runpod에서 google drive 연결 시 오류 발생
0
128
2
로드맵 선택
0
74
1
mmcv
0
66
2
Anchor box의 Positive 처리 위치
0
71
2
해당 강의 runpod 적용 후 에러 제보드립니다
0
96
2
run pod credit 관련 제보
0
126
2
mmdetection 2.x과 3.x 호환 관련 표기
0
89
2
mm_faster_rcnn_train_kitti.ipynb 실행 오류
0
114
3
질문 드립니다.
0
89
3
mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.
0
89
1
강사님께 수정을 제안드리고 싶은 것이 있습니다.
0
102
1
google automl efficientdet 다운로드 및 설치 오류
0
87
1
이상 탐지에 사용할 비전 기술 조언 부탁드립니다.
0
111
2
OpenCV 관련 질문드립니다.
0
87
2
mmcv 설치관련해서 문의드려요
0
357
3
강의 구성 관련해서 질문이 있습니다
1
141
2
모델 변환 성능 질문드립니다.
0
126
1