未验证 提交 a1f49a70 编写于 作者: L littletomatodonkey 提交者: GitHub

Update predict_system.py

上级 2f1bd9ab
...@@ -82,8 +82,7 @@ class SystemPredictor(object): ...@@ -82,8 +82,7 @@ class SystemPredictor(object):
output = [] output = []
results = self.det_predictor.predict(img) results = self.det_predictor.predict(img)
for result in results: for result in results:
#print(result) xmin, ymin, xmax, ymax = result["bbox"].astype("int")
xmin, xmax, ymin, ymax = result["bbox"].astype("int")
crop_img = img[xmin:xmax, ymin:ymax, :].copy() crop_img = img[xmin:xmax, ymin:ymax, :].copy()
rec_results = self.rec_predictor.predict(crop_img) rec_results = self.rec_predictor.predict(crop_img)
result["featrue"] = rec_results result["featrue"] = rec_results
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册