From fd49465754d8d1a691c6eb57afe96836811d6af6 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Thu, 20 May 2021 18:24:34 +0800 Subject: [PATCH] correct min_subgraph_size in ttfnet (#3088) --- deploy/python/infer.py | 4 ++-- ppdet/engine/export_utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 47a18d800..cd8721ddf 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -123,14 +123,14 @@ class Detector(object): MaskRCNN's results include 'masks': np.ndarray: shape: [N, im_h, im_w] ''' - self.det_times.postprocess_time_s.start() + self.det_times.preprocess_time_s.start() inputs = self.preprocess(image) np_boxes, np_masks = None, None input_names = self.predictor.get_input_names() for i in range(len(input_names)): input_tensor = self.predictor.get_input_handle(input_names[i]) input_tensor.copy_from_cpu(inputs[input_names[i]]) - self.det_times.postprocess_time_s.end() + self.det_times.preprocess_time_s.end() for i in range(warmup): self.predictor.run() output_names = self.predictor.get_output_names() diff --git a/ppdet/engine/export_utils.py b/ppdet/engine/export_utils.py index a317d45e0..69ac1bfc1 100644 --- a/ppdet/engine/export_utils.py +++ b/ppdet/engine/export_utils.py @@ -34,7 +34,7 @@ TRT_MIN_SUBGRAPH = { 'S2ANet': 40, 'EfficientDet': 40, 'Face': 3, - 'TTFNet': 3, + 'TTFNet': 60, 'FCOS': 16, 'SOLOv2': 60, 'HigherHRNet': 3, -- GitLab