첫 번째는 submission.to_csv('submission_ensemble.csv',index=None) 에서 index = None으로 바꿔주셔야 맨 왼쪽 index가 생성이 안됩니다.
두 번째는 prediction_string += str(int(label)) + ' ' + str(score) + ' ' + str(box[0] image_info['width']) + ' ' + str(box[1] image_info['height']) + ' ' + str(box[2] image_info['width']) + ' ' + str(box[3] image_info['height']) + ' ' 이 부분에서 prediction_string += str(int(label)) 을 해주셔야 class label이 7.0이 아니고 7 같은 int 자료형으로 바뀝니다. 이 두 부분 유의해서 ensemble 시켜주시면 될 것 같습니다.
세 번째는 혹시 wbf를 적용하실 때에는 weighted_boxes_fusion(boxes_list, scores_list, labels_list, iou_thr=iou_thr,skip_box_thr=skip_box_thr) nms 대신 이것만 추가해주시고 ski_box_thr 는 하이퍼 파라미터로 설정해주시면 됩니다.
개요
기본 baseline으로 모델 앙상블 할 때에 조심해야 할 점이 2가지 있습니다.
내용