ssd 학습 결과 궁금증
안녕하세요, 선생님. 제가 너무 설명이 부족했던 것 같습니다. Tensorflow object detection API를 이용하여 custom data로 학습을 진행하고 있습니다. (맞습니다.. 개정판 강의에서는 커스텀 데이터로 학습하는 강의가 없습니다..하하..^^;;) 현재 학습된 데이터는 제가 학습을 진행하지 않아서 train 몇 장, val 몇 장인지는 정확하게 모르겠습니다. 들은 바로는 40만장?으로 이미지 수는 충분한 듯 보였습니다. 그리고 person 클래스 1개만 설정하고 학습을 진행했습니다. 사람을 잡고는 있지만 background 또한 사람으로 인식하고 있는데 과하게 background를 많이 잡는 모습을 보여주고 있습니다. 처음에는 단순히 overfitting하게 학습을 진행했거나 아니면 데이터가 제대로 정리되어있지 않거나 또는 모델 파라미터가 잘 못 되었지 않았을까 하고 생각했습니다. 막상 train 1000장 val 250장으로 테스트해본 결과, background를 많이 잡는 것을 볼 수 있었습니다. yolo 같은 경우에는 데이터가 부족하여도 사람을 인식률이 떨어졌지 background를 계속 사람으로 잡고 있지는 않았던 것으로 기억하고 있습니다. 생각보다 SSD는 이상하게도 사람과 background를 골고루 잡고 있는 모습이 보였습니다. 현재 background를 사람으로 잡는(오탐지)를 후처리 전에 최대한 줄이고자 진행하고 있습니다. 아래 부분은 제가 많이 부족하여 사진(tensorboard 내용)으로 남겼습니다. mAP, 또는 다른 평가 지표가 클래스 별로 어떻게 되는 건지 자세한 정보 부탁드립니다. 꼭 해답이 아니더라도 첨언을 해주신다면 정말 감사드립니다. (사진) (사진)(사진)