diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 47a18d8005e2f80edda85e23f5e11de5f0acc682..cd8721ddf78854959f8df965458a95b39f52aab2 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 a317d45e0f58fee999533d19e70c7f2a00c0f139..69ac1bfc1a03fd5d76e9ffa3482ace0ece99d36f 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,