未验证 提交 fd494657 编写于 作者: W wangguanzhong 提交者: GitHub

correct min_subgraph_size in ttfnet (#3088)

上级 b274f7ad
......@@ -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()
......
......@@ -34,7 +34,7 @@ TRT_MIN_SUBGRAPH = {
'S2ANet': 40,
'EfficientDet': 40,
'Face': 3,
'TTFNet': 3,
'TTFNet': 60,
'FCOS': 16,
'SOLOv2': 60,
'HigherHRNet': 3,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册