|
@@ -323,7 +323,7 @@ class BatchAnalyze:
|
|
|
layout_res_item['poly'][4], layout_res_item['poly'][5]]
|
|
layout_res_item['poly'][4], layout_res_item['poly'][5]]
|
|
|
layout_res_width = layout_res_bbox[2] - layout_res_bbox[0]
|
|
layout_res_width = layout_res_bbox[2] - layout_res_bbox[0]
|
|
|
layout_res_height = layout_res_bbox[3] - layout_res_bbox[1]
|
|
layout_res_height = layout_res_bbox[3] - layout_res_bbox[1]
|
|
|
- if ocr_text in ['(204号', '(20', '(2', '(2号'] and ocr_score < 0.8 and layout_res_width < layout_res_height:
|
|
|
|
|
|
|
+ if ocr_text in ['(204号', '(20', '(2', '(2号', '(20号'] and ocr_score < 0.8 and layout_res_width < layout_res_height:
|
|
|
layout_res_item['category_id'] = 16
|
|
layout_res_item['category_id'] = 16
|
|
|
|
|
|
|
|
total_processed += len(img_crop_list)
|
|
total_processed += len(img_crop_list)
|