- 동물 병원 방문이 쉽지 않은 일반 사용자들에게 스마트폰으로 촬영한 사진으로 간편하게 안구, 피부 질병에 대해 진단 받아 반려동물의 건강을 집에서도 예측하고 관리할 수 있다.
- 반려동물을 디즈니 공주 캐릭터로 바꿔주는 생성 모델을 사용해볼 수 있다.
| 김용우 | 박종서 | 서영덕 | 신현준 | 조수혜 |
|---|---|---|---|---|
 |
 |
 |
 |
 |
| GitHub | GitHub | GitHub | GitHub | GitHub |
├── DualStyleGAN
├── api
│ └── app.py
├── demo : streamlit 코드와 front에 사용한 이미지
├── src : 모델 학습을 위한 코드 => lightning + hydra zen
│ ├── __init__.py
│ ├── config
│ ├── data
│ ├── loss
│ ├── model
│ ├── module
│ ├── utils
│ └── train.py
├── worker : 서버들간 연결을 위해 사용한 backend 코드
│ ├── utils : gradcam을 사용하기 위함
│ │ ├── __init__.py
│ │ └── gradcam.py
│ ├── __init__.py
│ ├── cat_skin_worker.py
│ ├── cat_eyes_worker.py
│ ├── dog_skin_worker.py
│ ├── dog_eyes_worker.py
│ └── gan_model_worker.py
├── web : 웹페이지를 구성하기 위해 사용한 frontend 코드
├── poetry.toml
├── pyproject.toml
├── run.py
└── .gitignore
- 강아지 피부 : ConvNext(encoder), Cascade-RCNN(decoder)
- 고양이 피부 : MANet(encoder), HRNet(decoder)
- 강아지 안구 : ResNest50
- 고양이 안구 : ResNest50
- 안구 detection : Yolov3(backbone: darknet53)
- 생성 모델 : DualStyleGAN
- pytorch lightining : https://github.com/Lightning-AI/lightning
- hydra-zen : https://github.com/mit-ll-responsible-ai/hydra-zen
- mmdetection : https://github.com/open-mmlab/mmdetection
- segmentation_models_pytorch : https://github.com/qubvel/segmentation_models.pytorch
- huggingface : https://huggingface.co/
- dual style gan : https://github.com/williamyang1991/DualStyleGAN
- AI hub : https://aihub.or.kr/