tritonuas/computer-vision-server repository overview

⁠computer-vision-server

(open for name suggestions 👀)

A flask server that automates all the tasks in our computer vision pipeline.

Designed to interface with Hub⁠ (ground control station backend)

⁠Setup

⁠Dependencies

• pipenv⁠ (with pyenv⁠)
• Docker (soon)

⁠Packages

pipenv install

⁠Directories

Create these directories in the project root:

• images/raw
• models/saliency
• models/segmentation
• models/classification

⁠Modules

TBD

⁠Models

• Saliency model
  1. Download trained model from our google drive⁠ or train a model yourself from here⁠.
  2. Place model in the "models/saliency" folder
• Segmentation model
  • TBD
• Classification model
  • TBD

⁠Configs

TBD

⁠Usage

pipenv run main

⁠Hub Connection

Example metadata hub needs to upload with each image to /upload:

{
  "filename": "",
  "longitude": "",
  "latitude": "",
  "altitude": "",
  "heading": "",
  "time": "",
  "groundspeed": "",
  "airspeed": "",
  "roll": "",
  "pitch": "",
  "yaw": ""
}