이미지인식1 AI 기반 아동 미술심리 진단을 위한 그림 분석 - YOLOv5(최종) 두 개의 게시물을 통해 여러 모델을 이용해서 이미지 디텍션을 진행했다. 이번에는 Yolov5를 이용해서 '여자 사람'에 대해 이미지 디텍션 모델 학습을 진행했다.현재까지 학습에 사용한 모델은 '나무', '집'은 Faster R-CNN을 이용해서 이미지 디텍션을 진행했다. 그리고 '남자 사람'은 SSD를 이용해서 이미지 디텍션을 진행했다. 그리고 이번에 이용한 Yolov5를 이용해서 '여자 사람'에 대해서 이미지 디텍션을 진행했다. Faster R-CNN, SSD, Yolov5 모두 사전 훈련된 가중치를 모델에 이용했다. 하지만 모델의 모든 가중치가 학습 과정에서 업데이트되도록 설정했다. 또한, Faster R-CNN과 SSD는 MMDetection을 사용하여 COCO 데이터셋 형식으로 처리하였지만, YO.. 2024. 12. 17. 이전 1 다음 728x90
