합성데이터 기반 객체 탐지 AI 경진대회

yolov8 사용해서 train 했는데요

2023.06.10 01:58 2,535 Views

어느정도 학습시켜서 test데이터로 예측 진행하려는데

일단 test데이터의 png파일이 3400개인데 yolov8 predict사용해서 txt파일 만들면 3378개가 나오는데..

다른분들도 이런가요? 이거 어느부분에서 손봐야할지...

로그인이 필요합니다
0 / 1000
ili0820
2023.06.10 12:59

1. test 데이터를 전부 제대로 load 하지 않고 일부 데이터에 대해서만 predict 하였을 수도 있습니다. 데이터가 전부 잘 불러와졌는지 확인이 필요합니다.
2. 학습한 모델이 일부 이미지에서 물체를 검출하지 못하였을 수 있습니다. 혹은 이미지에 물체가 없었거나, confidence 값 등으로 threshold를 설정하셨다면 너무 높은 값으로 설정하셨다든지 여러 가지 이유가 있을 수 있습니다. 
여러 가지 문제점을 고민해 보시는 게 좋을 듯합니다.

0liver
2023.06.11 00:43

컨피던스 IOU 적용 후 탐지된 물체가 없는 사진이 22개 있는 것 같네요