신호등 검출 AI대회 - (5) Data Imbalancing

[정다훈]

# 데이터 불균형 문제

데이터 분포도

훈련중에 사용한 데이터갯수를 그래프화 했는데 특정 데이터만 너무 많아서 데이터가 적은것은 잘 학습이 되지 않는다는것을 알게되었고, loss함수에서 특정클래스의 weight값을 증가시켜 학습이 더 잘되도록 하게 하려고 했는데 yolo 함수의 파라미터에서는 그런 값을 조정하는 파라미터가 없다는것을 알게 되었다.

https://github.com/ultralytics/ultralytics/issues/2703

Is it possible to change class weight? · Issue #2703 · ultralytics/ultralytics

yolo Ai관계자가 써놓은 답변

<yolo에서 data imbalancing 해결하는 방법>

https://y-t-g.github.io/tutorials/yolo-class-balancing/

Balance Classes During YOLO Training Using a Weighted Dataloader