tf_model_api module

Class definitions for YOLO model and Red-Blue line detection model

class tf_model_api.LineDetector(input_size=[500, 100], numClasses=4, weightsPath='tensorflow-yolov3/models/Flu_audere_line/1')

Bases: object

wrapper(image)

Wrapper method for the whole service for the Line Detection service

Parameters

image (numpy.array) – BGR image

Returns

Probability of detection list: Y-axis of predicted point numpy.ndarray: Image cropped

Return type

list

class tf_model_api.YOLO(input_size=512, numClasses=4, weightsPath='tensorflow-yolov3/models/Flu_audere/1')

Bases: object

wrapper(image)

Wrapper method for the whole service for the object Detection service

Parameters

image (numpy.ndarray) – BGR image

Returns

Bounding boxes per category of class detected with confidence score

Return type

list