Skip to content

Commit fd49465

Browse files
authored
correct min_subgraph_size in ttfnet (#3088)
1 parent b274f7a commit fd49465

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

deploy/python/infer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,14 +123,14 @@ def predict(self, image, threshold=0.5, warmup=0, repeats=1):
123123
MaskRCNN's results include 'masks': np.ndarray:
124124
shape: [N, im_h, im_w]
125125
'''
126-
self.det_times.postprocess_time_s.start()
126+
self.det_times.preprocess_time_s.start()
127127
inputs = self.preprocess(image)
128128
np_boxes, np_masks = None, None
129129
input_names = self.predictor.get_input_names()
130130
for i in range(len(input_names)):
131131
input_tensor = self.predictor.get_input_handle(input_names[i])
132132
input_tensor.copy_from_cpu(inputs[input_names[i]])
133-
self.det_times.postprocess_time_s.end()
133+
self.det_times.preprocess_time_s.end()
134134
for i in range(warmup):
135135
self.predictor.run()
136136
output_names = self.predictor.get_output_names()

ppdet/engine/export_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
'S2ANet': 40,
3535
'EfficientDet': 40,
3636
'Face': 3,
37-
'TTFNet': 3,
37+
'TTFNet': 60,
3838
'FCOS': 16,
3939
'SOLOv2': 60,
4040
'HigherHRNet': 3,

0 commit comments

Comments
 (0)