objectdetection

Type Members

1. case class DecodeOutput() extends ImageProcessing with Product with Serializable

   Decode the detection output The output of the model prediction is a 1-dim tensor The first element of tensor is the number(K) of objects detected, followed by [label score x1 y1 x2 y2] * K For example, if there are 2 detected objects, then K = 2, the tensor may looks like 2, 1, 0.5, 10, 20, 50, 80, 3, 0.3, 20, 10, 40, 70 After decoding, it returns a 2-dim tensor, each row represents a detected object 1, 0.5, 10, 20, 50, 80 3, 0.3, 20, 10, 40, 70

2. case class ImInfo() extends ImageProcessing with Product with Serializable

   Generate imInfo imInfo is a tensor that contains height, width, scaleInHeight, scaleInWidth

3. sealed trait ObjectDetectorDataset extends AnyRef

4. case class ScaleDetection() extends ImageProcessing with Product with Serializable

   If the detection is normalized, for example, ssd detected bounding box is in [0, 1], need to scale the bbox according to the original image size.

5. class Visualizer extends ImageProcessing

   used for image object detection visualize detected bounding boxes and their scores to image