object_detection_utils#

Classes

BoxList

ONNXRuntimeAdapterForFasterRCNN

TorchVision Faster RCNN model exported as ONNX

TorchVisionODAdapter

Adapter for interfacing with TorchVision's object detection models.

Functions

collate_fn(data)

compute_coco_eval(outputs, targets, ...)

Return mAP averaged over 0.5-0.95 using pycocotools eval.

draw_boxes(x, y, class_names, class_colors)

Given an image and a BoxList, draw the boxes in the BoxList on the image.

get_coco_gt(targets, num_class_ids)

get_coco_preds(outputs)