Module yolo

Module yolo 

Source
Expand description

Utility functions for a yolo model.

FunctionsΒ§

detect
Performs the actual detection and outputs the results.
get_env πŸ”’
Fetches an environment variable, or returns a default value if it is not set.
intersection πŸ”’
Finds the overlapping area between two bounding boxes. Returns a negative value if the boxes don’t overlap.
load_model
Loads a YOLO model from the path specified by the environment variable SAURON_MODEL_PATH.
post_process πŸ”’
Performs non-maximum suppression on the detections.
pre_process πŸ”’
Pre-processes an image for YOLO detection.
union πŸ”’
Finds the total area covered by both bounding boxes (But breaks if the boxes don’t overlap).