erikgartner/wara-ps-detector repository overview

⁠Detector

The detector module for the WARA PS simulator

⁠Overview

This module exposes a REST API that allows other modules to submit images for analyzing.

Run the module with the command:

python -m detector.main

⁠Installation

Install Python 3.6 and the Python packages in requirements.txt.

pip3 install -r requirements.txt

Then download all weights using:

cd models/weights; bash install.sh

As an alternative a prebuilt Docker image is available as:

# CPU version
docker pull erikgartner/wara-ps-detector:latest

# GPU verson
docker pull erikgartner/wara-ps-detector:gpu

⁠API

The API documentation is available through Postman⁠.

The endpoints are:

• GET / - module information
• POST /detect - uploads image to analyze