boostcampaitech7 / level2-objectdetection-cv-18

level2-objectdetection-cv-18 created by GitHub Classroom
0 stars 3 forks source link

버그: ensemble.py #86

Closed nOctaveLay closed 1 month ago

nOctaveLay commented 1 month ago
main()

File "/data/ephemeral/home/chan/level2-objectdetection-cv-18/tools/ensemble_1.py", line 125, in main boxes, scores, labels = make_ensemble_format_per_image(image_id, output, image_width = image_width, image_height = image_height) File "/data/ephemeral/home/chan/level2-objectdetection-cv-18/tools/ensemble_1.py", line 38, in make_ensemble_format_per_image predict_string = csv_data[csv_data['image_id'] == image_id]['PredictionString'].tolist()[0] IndexError: list index out of range

문제 원인 : 모든 모델이 한 이미지에 대해서 bounding box를 예측하지 못했을 때 생기는 문제 해결 : 모델이 bounding box를 예측하지 못하면 csv_data[csv_data['image_id'] == image_id]['PredictionString'].tolist() 의 list는 비어있게 된다. 이를 바탕으로 만약 모든 모델이 bounding box를 해결하지 못하면 아예 그 부분은 건너뛰기로 했다.