correct min_subgraph_size in ttfnet (#3088)

PaddlePaddle · May 20, 2021 · fd49465 · fd49465
1 parent b274f7a
commit fd49465
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/deploy/python/infer.py b/deploy/python/infer.py
@@ -123,14 +123,14 @@ def predict(self, image, threshold=0.5, warmup=0, repeats=1):
                             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
@@ -34,7 +34,7 @@
     'S2ANet': 40,
     'EfficientDet': 40,
     'Face': 3,
-    'TTFNet': 3,
+    'TTFNet': 60,
     'FCOS': 16,
     'SOLOv2': 60,
     'HigherHRNet': 3,