From 872ed91d8a89ebb929afb9ea693721a06d6d2d28 Mon Sep 17 00:00:00 2001 From: Olivia Jain Date: Mon, 4 Apr 2022 13:23:25 -0700 Subject: [PATCH] Perf FasterRCNN + MaskRCNN (#11102) * add faster mask * fix paths --- .../python/tools/tensorrt/perf/model_list.json | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/tensorrt/perf/model_list.json b/onnxruntime/python/tools/tensorrt/perf/model_list.json index 9b87cee0df..c78b34435f 100644 --- a/onnxruntime/python/tools/tensorrt/perf/model_list.json +++ b/onnxruntime/python/tools/tensorrt/perf/model_list.json @@ -1,4 +1,16 @@ [ + { + "model_name": "FasterRCNN-10", + "working_directory": "./onnx-zoo-models/FasterRCNN-10", + "model_path": "./faster_rcnn_R_50_FPN_1x.onnx", + "test_data_path": "./" + }, + { + "model_name": "MaskRCNN-10", + "working_directory": "./onnx-zoo-models/MaskRCNN-10/", + "model_path": "./mask_rcnn_R_50_FPN_1x.onnx", + "test_data_path": "./" + }, { "model_name": "mobilenetv2-7", "working_directory": "./onnx-zoo-models/mobilenetv2-7/", @@ -221,4 +233,4 @@ "model_path": "./mosaic/mosaic.onnx", "test_data_path": "./mosaic/" } -] \ No newline at end of file +]