AIHub에서 제공하는 데이터를 사용했습니다.
Flammable Object, 즉 가연성 물체를 크게 비닐류, 종이류, 페트병류, 플라스틱류 4가지로 분류 했습니다.
y_pred.csv - 제출 시 사용할 수 있는 csv 파일입니다.
학습 시킬 데이터 셋은 git에 올렸습니다. 베이스라인에 따라 실행시키면 DataSet을 바로 불러올 수 있습니다.
images
train
valid
test
labels
train
valid
test
train, valid, test 각각의 데이터 셋의 비율은 8:1:1입니다.
사진 업로드 용량에 대한 시간적인 제약이 있어 train(240장), valid(30장), test(30장) 으로 진행합니다.
예시 이미지는 학습된 것을 기반으로 테스트한 결과입니다.

