Improved model support - Githubissues

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT

BSD 3-Clause "New" or "Revised" License

682 stars 245 forks source link

Closed ghost closed 5 years ago

ghost commented 5 years ago

support for batch size > 1
support for ssd_mobilenet_v2
changed names from scores, boxes, classes -> detection_scores, detection_boxes, detection_classes
added NMS score threshold to build_detection_graph