未验证 提交 04cde45e 编写于 作者: B Bin Lu 提交者: GitHub

Update recognition_web_service.py

上级 a64c1238
...@@ -83,7 +83,7 @@ class DetOp(Op): ...@@ -83,7 +83,7 @@ class DetOp(Op):
} }
return feed_dict, False, None, "" return feed_dict, False, None, ""
def postprocess(self, input_dicts, fetch_dict, log_id): def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
boxes = self.img_postprocess(fetch_dict, visualize=False) boxes = self.img_postprocess(fetch_dict, visualize=False)
boxes.sort(key=lambda x: x["score"], reverse=True) boxes.sort(key=lambda x: x["score"], reverse=True)
boxes = filter(lambda x: x["score"] >= self.threshold, boxes = filter(lambda x: x["score"] >= self.threshold,
...@@ -173,7 +173,7 @@ class RecOp(Op): ...@@ -173,7 +173,7 @@ class RecOp(Op):
filtered_results.append(results[i]) filtered_results.append(results[i])
return filtered_results return filtered_results
def postprocess(self, input_dicts, fetch_dict, log_id): def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
batch_features = fetch_dict["features"] batch_features = fetch_dict["features"]
if self.feature_normalize: if self.feature_normalize:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册