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

알고리즘 | 비전 | 분류 | Accuracy

상금 : 1,000 만원
946명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

privat 6th

Redix6

2023.01.31 04:59 1,226 조회 language

제출 안하려고 했는데, 그냥 PPT 연습 겸 코드도 공유해볼 겸 올립니다.

전처리는 colab에서 진행하였습니다.
따로 진행한 부분들이 있는데 올리느랴 전부 합쳤습니다. 동작하는데는 이상 없을 것으로 생각됩니다.
.py파일이 안올라가서 부득이하게 ipynb 노트북 파일의 한 셀에다가 전부 붙여 넣었습니다.
.py파일을 생성해서 복사 붙여넣기 하시면 됩니다.

Preprocessing : https://colab.research.google.com/drive/1mKDS7ExmmUpo-2nLmnxvWEzDuy7i13E5?usp=sharing
versions :
python==3.8.16
albumentations==1.3.0
opencv-python==4.7.0.68
torch==1.12.1
timm==0.6.12
numba==0.56.4

파일 폴더 구조:
test
train
cleaned2
--train_prep
--test_prep
run(아무폴더)
--run.py (메인코드)
train.csv
test.csv
samplesubmission.csv