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

Merge branch 'dygraph' into trt_cpp

...@@ -76,7 +76,7 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes, ...@@ -76,7 +76,7 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,
float(*std::max_element(&predict_batch[n * predict_shape[2]], float(*std::max_element(&predict_batch[n * predict_shape[2]],
&predict_batch[(n + 1) * predict_shape[2]])); &predict_batch[(n + 1) * predict_shape[2]]));
if (argmax_idx > 0 && (!(i > 0 && argmax_idx == last_index))) { if (argmax_idx > 0 && (!(n > 0 && argmax_idx == last_index))) {
score += max_value; score += max_value;
count += 1; count += 1;
str_res.push_back(label_list_[argmax_idx]); str_res.push_back(label_list_[argmax_idx]);
......
...@@ -32,7 +32,6 @@ class MakeShrinkMap(object): ...@@ -32,7 +32,6 @@ class MakeShrinkMap(object):
text_polys, ignore_tags = self.validate_polygons(text_polys, text_polys, ignore_tags = self.validate_polygons(text_polys,
ignore_tags, h, w) ignore_tags, h, w)
gt = np.zeros((h, w), dtype=np.float32) gt = np.zeros((h, w), dtype=np.float32)
# gt = np.zeros((1, h, w), dtype=np.float32)
mask = np.ones((h, w), dtype=np.float32) mask = np.ones((h, w), dtype=np.float32)
for i in range(len(text_polys)): for i in range(len(text_polys)):
polygon = text_polys[i] polygon = text_polys[i]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册