🏆 마스크 착용 상태 분류 Competition

COVID-19의 확산으로 올바른 마스크 착용 상태를 검사하기 위해 카메라로 비춰진 사람 얼굴 이미지가 마스크를 쓰고 있는지, 쓰지 않았는지, 정확히 쓴 것인지 자동으로 가려낼 수 있는 시스템이 필요

🎬 학습데이터

• 전체 사람 수 : 4500
• 한 사람당 사진의 개수 : 7 [[마스크 착용 5장, 이상하게 착용(코스크, 턱스크) 1장, 미착용 1장]]
• 이미지 크기: (384, 512)

🌳 구조

Train.py

• 데이터셋을 읽고 전처리를 진행한 후 데이터를 하나씩 꺼내주는 Dataset 클래스를 구현한 파일

Inference.py

• 학습 완료된 모델을 통해 test set 에 대한 예측 값을 구하고 이를 .csv 형식으로 저장하는 파일

Dataset.py

• 마스크 데이터셋을 읽고 전처리를 진행한 후 데이터를 하나씩 꺼내주는 Dataset 클래스를 구현한 파일

Model.py

• 데이터를 받아 연산을 처리한 후 결과 값을 내는 Model 클래스를 구현하는 파일

Loss.py

• 이미지 분류에 사용될 수 있는 다양한 Loss 들을 정의한 파일