마스크 착용 여부, 연령대, 성별에 따라 총 18개 클래스로 이미지 분류하는 AI 알고리즘 개발
본 대회에 대한 세부적인 내용은 아래 링크를 통해 확인하실 수 있습니다.
그림과 같은 구조로 앞단의 레이어들은 동일한 네트워크를 사용하여 파라미터를 공유하도록 만들었고, 뒷단의 레이어들은 각각의 문제(나이, 성별, 마스크 착용 여부)를 풀기 위한 독립된 네트워크로 구현하였습니다.
아래 저장소에서 모델 코드를 자세히 확인하실 수 있습니다.
https://github.com/khyeyoon/Head-Sharing-Multi-Class-Classification
cd src
python train.py