boostcampaitech3 / final-project-level3-cv-09

아동 및 심약자를 위한 동영상 폭력성 감지 및 필터링 서비스
5 stars 6 forks source link

[Model] Evaluate Yolov5 #81

Open tjrudrnr2 opened 2 years ago

tjrudrnr2 commented 2 years ago

Why?

Determine why blood detection performance is low

  1. Object Detection or Segmentation
  2. Lack of Data
  3. Etc

    How?

    Visualize mAP, loss and inference result

tjrudrnr2 commented 2 years ago

image image

training data : 약 550개, validation data : 약 150개 데이터 수가 적어 overfitting 우려가 있기 때문에 epoch 30으로 학습하였다.

  1. default
  2. augmentation p = 0.01 -> 0.1
  3. augmentation p = 0.01 -> 0.1, epoch 60 => 데이터가 적어서 그런가 애초에 학습이 너무 불안정하고 epoch가 30보다 클 경우 mAP 상승폭도 미비하고 loss는 학습이 이루어지지 않는 듯 하다. 그리고 loss 는 왜 저리 낮게 나오는지 모르겠다.
jeongjae96 commented 2 years ago

데이터를 더 모아야겠네요!

tjrudrnr2 commented 2 years ago

image image

augmentation이 들어간 default랑 아예 없는애랑 유의미한 차이는 보이지 않는다.

tjrudrnr2 commented 2 years ago

image image

Albumentation의 하나인 RGBShift()을 추가해도 큰 차이는 없다. 더 확인해볼게 있을까요?