From 90a94f3147a7cecfd0a8b1b60bf6207184d1f128 Mon Sep 17 00:00:00 2001 From: Marco Salathe Date: Thu, 18 Oct 2018 10:37:37 -0700 Subject: [PATCH] Added ssdlite_mobilenet_v2_coco --- tf_trt_models/detection.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tf_trt_models/detection.py b/tf_trt_models/detection.py index f1341fc..cbd57ee 100644 --- a/tf_trt_models/detection.py +++ b/tf_trt_models/detection.py @@ -38,6 +38,11 @@ 'http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v2_coco_2018_03_29.tar.gz', 'ssd_mobilenet_v2_coco_2018_03_29', ), + 'ssdlite_mobilenet_v2_coco': DetectionModel( + 'ssdlite_mobilenet_v2_coco', + 'http://download.tensorflow.org/models/object_detection/ssdlite_mobilenet_v2_coco_2018_05_09.tar.gz', + 'ssdlite_mobilenet_v2_coco_2018_05_09', + ), 'ssd_inception_v2_coco': DetectionModel( 'ssd_inception_v2_coco', 'http://download.tensorflow.org/models/object_detection/ssd_inception_v2_coco_2018_01_28.tar.gz',