건설용 자갈 암석 종류 분류 AI 경진대회

CNN 복습할 겸 오늘 대회에 한번 참여해 봤는데,
데이터 양도 많고 나름 재밌을 것 같아서 이렇게 제 코드를 올려봅니다.

최대한 주석으로 처리했으니, 이해해 보시고 이해 안되시면 댓글 남겨주세요!
사실상 건드릴 부분은 거의 마지막 부분의 "최종 함수/Config"부분 입니다.
Sequence Data 기반으로 augmentor도 적용할 수 있습니다. 여기서는 albumentation이 적용된다고 가정했는데 원치 않으시면 None으로 처리해주시면 됩니다.

함수 전체 복사 하셔서 마지막 config만 건드리시고 end_to_end()함수를 실행해 주시면 됩니다.(물론, 모든 함수 다 실행 되어있어야 합니다.)
모델은 최대한 다양하게 넣어보려고 노력했는데 앞서 저보다 훌륭하신 분께서 "큰 모델이 성능이 좋은 것 같다"라고 하셔서 heavy한 모델도 넣어봤습니다.

대충 xception 적용시 epoch10정도 돌리면, 0.7정도가 나오네요.(augmentation 적용 안 함.)

참고로 맨 마지막 Predict and Submit부분은 test_ds가 있어야지 실행됩니다.(local에서 따로 실행해야됩니다. "Create_Sequence_DataSet()"이 함수에요.)

+ 좀 더 개선해본 결과 아래와 같은 코드로 Config 수정했고 성능은 Public 기준 0.82정도가 나옵니다. 좀 더 자세한 내용은 토크에 올린 제 게시물 확인 바랍니다.😊