Segmentation Model Serving Basic

Streamlit과 FastAPI로 만든 간단한 Hanbone Segmentation 페이지

사용 방법

Frontend

cd frontend
poetry run streamlit run ui.py

Backend

cd backend
poetry run uvicorn server:app --reload

사용 예시

구조도

.
├── README.md
├── backend
│   ├── segmentation_module.py
│   └── server.py
├── frontend
│   └── ui.py
├── models
│   ├── resnet-fpn.pt
│   └── swin-fpn.pt
├── poetry.lock
└── pyproject.toml

사용 가능 모델

Encoder

• Swin Transformer - B [paper]

• ResNet 101 [paper]

Decoder

• Feature Pyramid Networks(FPN) [paper]

참고자료

https://github.com/davidefiocco/streamlit-fastapi-model-serving.git