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

correct min_subgraph_size in ttfnet (#3088)

上级 b274f7ad
...@@ -123,14 +123,14 @@ class Detector(object): ...@@ -123,14 +123,14 @@ class Detector(object):
MaskRCNN's results include 'masks': np.ndarray: MaskRCNN's results include 'masks': np.ndarray:
shape: [N, im_h, im_w] shape: [N, im_h, im_w]
''' '''
self.det_times.postprocess_time_s.start() self.det_times.preprocess_time_s.start()
inputs = self.preprocess(image) inputs = self.preprocess(image)
np_boxes, np_masks = None, None np_boxes, np_masks = None, None
input_names = self.predictor.get_input_names() input_names = self.predictor.get_input_names()
for i in range(len(input_names)): for i in range(len(input_names)):
input_tensor = self.predictor.get_input_handle(input_names[i]) input_tensor = self.predictor.get_input_handle(input_names[i])
input_tensor.copy_from_cpu(inputs[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): for i in range(warmup):
self.predictor.run() self.predictor.run()
output_names = self.predictor.get_output_names() output_names = self.predictor.get_output_names()
......
...@@ -34,7 +34,7 @@ TRT_MIN_SUBGRAPH = { ...@@ -34,7 +34,7 @@ TRT_MIN_SUBGRAPH = {
'S2ANet': 40, 'S2ANet': 40,
'EfficientDet': 40, 'EfficientDet': 40,
'Face': 3, 'Face': 3,
'TTFNet': 3, 'TTFNet': 60,
'FCOS': 16, 'FCOS': 16,
'SOLOv2': 60, 'SOLOv2': 60,
'HigherHRNet': 3, 'HigherHRNet': 3,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册