diff --git a/modelscope/pipelines/cv/ocr_utils/model_dla34.py b/modelscope/pipelines/cv/ocr_utils/model_dla34.py index cd114a48..9f2b6396 100644 --- a/modelscope/pipelines/cv/ocr_utils/model_dla34.py +++ b/modelscope/pipelines/cv/ocr_utils/model_dla34.py @@ -635,7 +635,7 @@ class DLAUp(nn.Module): "ida_{}".format(i), IDAUp(3, channels[j], in_channels[j:], scales[j:] // scales[j]), ) - scales[j + 1 :] = scales[j] + scales[j + 1:] = scales[j] in_channels[j + 1 :] = [channels[j] for _ in channels[j + 1 :]] def forward(self, layers): diff --git a/modelscope/pipelines/cv/ocr_utils/table_process.py b/modelscope/pipelines/cv/ocr_utils/table_process.py index 193f64ea..c8e91b11 100644 --- a/modelscope/pipelines/cv/ocr_utils/table_process.py +++ b/modelscope/pipelines/cv/ocr_utils/table_process.py @@ -212,7 +212,6 @@ def gbox_post_process(gbox, c, s, h, w): def nms(dets, thresh): if len(dets) < 2: return dets - scores = dets[:, 8] index_keep = [] keep = [] for i in range(len(dets)):