The backend consists of two containerized FastAPI services: the Image Feature Extraction Service (Port 8001) and the Screen Parsing Service (Port 8000).
Prerequisites
- Docker
- Docker Compose
- NVIDIA GPU support (requires NVIDIA Container Toolkit)
Model Weights Preparation
You must manually download model weights from OmniParser on Hugging Face and place them in the OmniParser/weights/ directory before starting, otherwise the API will fail to process images.
Required structure:
OmniParser/weights/icon_detect_v1_5/best.pt (YOLO detection model)OmniParser/weights/icon_caption_florence/ (Caption model files)
Deployment Commands
To build and start the services in the foreground:
docker-compose up --build
To run the services in the background:
docker-compose up -d --build