Skip to content

2021 NIPA 한국인 헤어스타일 경진대회 2등 솔루션 자료입니다.

Notifications You must be signed in to change notification settings

jyryu3161/Korean-Hair-Segmentation

 
 

Repository files navigation

💇 한국인 헤어 세그멘테이션 경진대회 2등 솔루션

Contents

Solution Description

Command Line Interface

  • Preprocess
  • Train
  • Predict

Solution Description

Subject

본 대회의 주제는 한국인 헤어에 대해서 세그멘테이션을 수행하는 AI 모델 개발입니다. 주어진 이미지는 아래의 그림과 같이 다양한 헤어스타일에 대해서 세그멘테이션을 수행해야 할 필요가 있었습니다.

task

Data

  • 학습 데이터: 약 20만장의 512x512 헤어스타일 이미지
  • 경진대회의 데이터의 경우 저작권의 문제때매, 예시 이미지의 경우 AI Hub에서 제공하는 한국인 헤어스타일 데이터셋의 예시를 사용했습니다.
  • 이미지 데이터 : https://aihub.or.kr/aidata/30752

Summary

summary

Score

image-20210914131303674

Code Structure

├── DATA
│   ├── Final_DATA
│   │   ├── task02_test
│   │   └── task02_train
│   ├── polygon_iou.json
│   └── data.csv
├── config
│   └── config.yaml
├── model
│   ├── submission_model_weight
│   └── preprocessing
│       └── preprocessing_model_weight
├── modules
│   ├── dataset.py
│   ├── models.py
│   ├── preprocessing.py
│   ├── scheduler.py
│   ├── solver.py
│   ├── transform.py
│   └── utils.py
├── output
│   ├── log
│   ├── runs
│   │   └── output_model_weight
│   └── result.json
├── README.md
├── predict.py
└── train.py

Command Line Interface

Preprocess

python preprocess.py

Train

python train.py

Predict

python predict.py

About

2021 NIPA 한국인 헤어스타일 경진대회 2등 솔루션 자료입니다.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%