SSD1 AI 기반 아동 미술심리 진단을 위한 그림 분석 - SSD512 이전 게시물에서 정확한 아동 미술심리 진단을 위한 그림을 진행하기 위해 Faster R-CNN을 이용해서 이미지 디텍션을 진행했다. 이번에는 SSD512를 이용해서 이미지 디텍션을 진행해보려고 한다.Faster R-CNN을 이용해서 '나무', '집' 두 개의 카테고리에 대해서 훈련을 진행했을 때 높은 정확도를 보여주었지만 시간이 오래 걸렸다. 반면 SSD512를 이용해 '남자 사람'을 훈련했을 때 훈련 시간이 기존 5시간 50분에서 4시간 10분으로 줄었다. 하지만 정확도에 있어서 상대적으로 낮은 정확도를 보여주었다. 또한, 사전 훈련된 모델을 사용했지만 아래와 같이 데이터 증강 부분은 수정해서 비활성화 하거나 최소화했다.cfg.data.train=dict( type='RepeatDatase.. 2024. 12. 6. 이전 1 다음 728x90
