mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update recognition_web_service.py

pull/1568/head
Bin Lu 2021-12-16 14:22:19 +08:00 committed by GitHub
parent a64c12385d
commit 04cde45eed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deploy/paddleserving/recognition/recognition_web_service.py
View File

@ -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):
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: