Training yolo model with multiple datasets

여러 데이터셋으로 한 모델을 학습시킬 때 한 데이터셋만으로 학습시킬 때보다 성능이 저하하는 현상을 겪고 있다.

 

게시글의 제목처럼 구글에 검색했고 참고할만한 자료를 아래에 정리해둔다. 곧 이를 참고해 해결법을 모색할 생각이다.

 

 

https://github.com/ultralytics/yolov5/issues/6781

What is the recommended way to train 2 new datasets with different classes via transfer learning in YOLOv5? · Issue #6781 · ul

https://github.com/ultralytics/yolov5/issues/9908

Training Multiple Datasets · Issue #9908 · ultralytics/yolov5

위 두개의 질문글에서 관리자가 아래의 토론을 참고하라고 하였다.

 

 

 

https://community.ultralytics.com/t/how-to-combine-weights-to-detect-from-multiple-datasets/38/6