🐾 HYPETCARE - 반려동물 질병 예측 서비스

• 동물 병원 방문이 쉽지 않은 일반 사용자들에게 스마트폰으로 촬영한 사진으로 간편하게 안구, 피부 질병에 대해 진단 받아 반려동물의 건강을 집에서도 예측하고 관리할 수 있다.
• 반려동물을 디즈니 공주 캐릭터로 바꿔주는 생성 모델을 사용해볼 수 있다.

👀 시연 영상

👨🏻‍💻 👩🏻‍💻 팀 구성

김용우	박종서	서영덕	신현준	조수혜
GitHub	GitHub	GitHub	GitHub	GitHub

📆 프로젝트 일정 : 2023.06.30 ~ 2023.07.28

📲 서비스 아키텍쳐

🍀 Folder Structure

├── 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

💫 Final Model

• 강아지 피부 : ConvNext(encoder), Cascade-RCNN(decoder)
• 고양이 피부 : MANet(encoder), HRNet(decoder)
• 강아지 안구 : ResNest50
• 고양이 안구 : ResNest50
• 안구 detection : Yolov3(backbone: darknet53)
• 생성 모델 : DualStyleGAN

🔍 Reference 및 출처

• 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/