有的图片报这个错误，怎么解决？ValueError: cannot select an axis to squeeze out which has size not equal to one #5

nissansz · 2023-12-13T23:03:45Z

nissansz
Dec 13, 2023

File "F:/pycharm2020.2/RapidStructure-0.0.0/TableStructureRec-main/demo_wired.py", line 31, in
table_str, elapse = table_rec(img_path)
File "F:\pycharm2020.2\RapidStructure-0.0.0\TableStructureRec-main\wired_table_rec\main.py", line 47, in call
polygons = self.table_line_rec(img)
File "F:\pycharm2020.2\RapidStructure-0.0.0\TableStructureRec-main\wired_table_rec\table_line_rec.py", line 69, in call
polygons = merge_adjacent_polys(polygons)
File "F:\pycharm2020.2\RapidStructure-0.0.0\TableStructureRec-main\wired_table_rec\utils_table_recover.py", line 79, in merge_adjacent_polys
polygons = combine_two_poly(polygons, idxs)
File "F:\pycharm2020.2\RapidStructure-0.0.0\TableStructureRec-main\wired_table_rec\utils_table_recover.py", line 88, in combine_two_poly
idxs = idxs.squeeze(0)
ValueError: cannot select an axis to squeeze out which has size not equal to one

原图

可视化表格框

qqeip · 2023-12-14T00:13:23Z

qqeip
Dec 14, 2023

这些都是阿里读光的表格识别模型

考虑到现有表格识别模型较弱，尝试阿里读光下的表格识别模型效果。
读光-表格结构识别-有线表格
 读光-表格结构识别-无线表格
如果效果满足，会将其整合到现有仓库下，欢迎有兴趣的小伙伴一起做。
详见 RapidAI/RapidOrientation#2

0 replies

nissansz · 2023-12-14T00:20:46Z

nissansz
Dec 14, 2023
Author

阿里demo结果也是一样的。你的现在的有线表格模型本身就是从阿里转换来的吧。
结果一样

0 replies

qqeip · 2023-12-14T00:22:30Z

qqeip
Dec 14, 2023

感觉是有个斜线

加个判断就可
def combine_two_poly(polygons: np.ndarray, idxs: np.ndarray) -> np.ndarray:
del_idxs, insert_boxes = [], []
# print(idxs)
if idxs.shape[1] == 1:
idxs = idxs.flatten()
else:
idxs = idxs.squeeze(0)

8 replies

qqeip Dec 14, 2023

你参考 lineless main.py
def call(self, content: Dict[str, Any]) -> str:
ss = time.perf_counter()
img = self.load_img(content)

    ocr_res, _ = self.ocr(img)

    input_info = self.preprocess(img)
    try:
        polygons, slct_logi = self.infer(input_info)
        logi_points = self.filter_logi_points(slct_logi)

        sorted_polygons = sorted_boxes(polygons)

        cell_box_map = match_ocr_cell(sorted_polygons, ocr_res)