포디블록 구조 추출 AI 경진대회

먼저 대회를 주최해주신 주최 측께 감사드립니다. 또 대회 참가하신 모든 분 고생 많으셨습니다!

저는 데이터 증강을 중점으로 이번 대회에 참가했습니다.
데이터셋의 배경을 랜덤한 이미지로 합성하고 두개의 블록사진을 의도적으로 겹처 구조가 잘 안보이는 블록의 경우에도 높을 정확도를 보일 수 있게 학습했습니다.
배경 합성의 경우 코드 공유 게시판의 coldjeans님의 코드를 참조했습니다!

py 파일이 업로드되지 않아 ipynb로 2가지 셀에 나눠 업로드합니다.
첫 번째에는 데이터 증강 코드, 두 번째에는 모델 학습 코드가 담겨있습니다.
첫 번째 데이터 증강의 경우 스레드를 사용해 실행시간을 단축했기 때문에 시드값과 상관없이 컴퓨터 환경에 따라 만들어진 데이터셋이 다를 수 있습니다.
때문에 Private의 모델을 구현하기 위해서는 아래 링크의 데이터셋을 다운받아 사용해 주세요.
리더보드의 모델의 경우 아래 모델 링크의 7번째 epoch를 학습한 모델로 구현할 수 있습니다.

폴더 구조는 아래와 같습니다.
indoorCVPR_09의 경우  PPT 맨 뒤쪽에 다운로드 링크 첨부했습니다.
./indoorCVPR_09/
./train/
./test/
./train.csv
./test.csv
./sample_submission.csv

데이터셋을 링크에서 받는 zip 파일을 같은 폴더에 압축해제 코드의 두번째 셀부터 실행해 주세요.

증강된 데이터셋 (배경제거 + 이미지겹치기) : https://konkukackr-my.sharepoint.com/:u:/g/personal/ptj0225_konkuk_ac_kr/Eb7RszQvaKxGmSeWN3MpynwBYTjrkYkG5CG59id05uTL9g?e=PFKjF9
모델 파라미터 : https://konkukackr-my.sharepoint.com/:u:/g/personal/ptj0225_konkuk_ac_kr/EevuK3gCsfdAkuQXicmaW2IB6jlU1KypviZNuSjq8GjKLQ?e=TLcyWL