add img count
Showing
2 changed files
with
26 additions
and
8 deletions
| ... | @@ -99,6 +99,7 @@ RES_FAILED = '识别失败' | ... | @@ -99,6 +99,7 @@ RES_FAILED = '识别失败' |
| 99 | RES_FAILED_1 = '识别失败(阶段1)' | 99 | RES_FAILED_1 = '识别失败(阶段1)' |
| 100 | RES_FAILED_2 = '识别失败(阶段2)' | 100 | RES_FAILED_2 = '识别失败(阶段2)' |
| 101 | RES_FAILED_3 = '识别失败(阶段1数据格式错误)' | 101 | RES_FAILED_3 = '识别失败(阶段1数据格式错误)' |
| 102 | RES_FAILED_SET = {RES_FAILED, RES_FAILED_1, RES_FAILED_2, RES_FAILED_3} | ||
| 102 | 103 | ||
| 103 | CARD_RATIO = 0.9 | 104 | CARD_RATIO = 0.9 |
| 104 | UNKNOWN_CARD = '未知卡号' | 105 | UNKNOWN_CARD = '未知卡号' | ... | ... |
| ... | @@ -310,13 +310,15 @@ class BSWorkbook(Workbook): | ... | @@ -310,13 +310,15 @@ class BSWorkbook(Workbook): |
| 310 | month_mapping.setdefault(item, []).append( | 310 | month_mapping.setdefault(item, []).append( |
| 311 | (ws.title, idx_list[i] + min_row, idx_list[i + 1] + min_row - 1, day_mean)) | 311 | (ws.title, idx_list[i] + min_row, idx_list[i + 1] + min_row - 1, day_mean)) |
| 312 | 312 | ||
| 313 | def build_metadata_rows(self, confidence, code, print_time, start_date, end_date): | 313 | def build_metadata_rows(self, confidence, code, print_time, start_date, end_date, res_count_tuple): |
| 314 | if start_date is None or end_date is None: | 314 | if start_date is None or end_date is None: |
| 315 | timedelta = None | 315 | timedelta = None |
| 316 | else: | 316 | else: |
| 317 | timedelta = (end_date - start_date).days | 317 | timedelta = (end_date - start_date).days |
| 318 | metadata_rows = [ | 318 | metadata_rows = [ |
| 319 | ('流水识别置信度', confidence), | 319 | ('流水识别置信度', confidence), |
| 320 | ('图片总数', res_count_tuple[0]), | ||
| 321 | ('识别成功', res_count_tuple[1]), | ||
| 320 | self.blank_row, | 322 | self.blank_row, |
| 321 | self.code_header, | 323 | self.code_header, |
| 322 | ] | 324 | ] |
| ... | @@ -330,8 +332,8 @@ class BSWorkbook(Workbook): | ... | @@ -330,8 +332,8 @@ class BSWorkbook(Workbook): |
| 330 | ) | 332 | ) |
| 331 | return metadata_rows | 333 | return metadata_rows |
| 332 | 334 | ||
| 333 | def build_meta_sheet(self, role_name, card, confidence, code, print_time, start_date, end_date): | 335 | def build_meta_sheet(self, role_name, card, confidence, code, print_time, start_date, end_date, res_count_tuple): |
| 334 | metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date) | 336 | metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date, res_count_tuple) |
| 335 | if not isinstance(role_name, str): | 337 | if not isinstance(role_name, str): |
| 336 | role_name = consts.UNKNOWN_ROLE | 338 | role_name = consts.UNKNOWN_ROLE |
| 337 | ms = self.create_sheet('{0}{1}({2})'.format(self.meta_sheet_title, role_name, card)) | 339 | ms = self.create_sheet('{0}{1}({2})'.format(self.meta_sheet_title, role_name, card)) |
| ... | @@ -595,7 +597,7 @@ class BSWorkbook(Workbook): | ... | @@ -595,7 +597,7 @@ class BSWorkbook(Workbook): |
| 595 | ms.append(row) | 597 | ms.append(row) |
| 596 | self.remove(tmp2_ws) | 598 | self.remove(tmp2_ws) |
| 597 | 599 | ||
| 598 | def bs_rebuild(self, bs_summary): | 600 | def bs_rebuild(self, bs_summary, res_count_tuple): |
| 599 | # bs_summary = { | 601 | # bs_summary = { |
| 600 | # '卡号': { | 602 | # '卡号': { |
| 601 | # 'classify': 0, | 603 | # 'classify': 0, |
| ... | @@ -656,7 +658,8 @@ class BSWorkbook(Workbook): | ... | @@ -656,7 +658,8 @@ class BSWorkbook(Workbook): |
| 656 | summary.get('code'), | 658 | summary.get('code'), |
| 657 | summary.get('print_time'), | 659 | summary.get('print_time'), |
| 658 | start_date, | 660 | start_date, |
| 659 | end_date) | 661 | end_date, |
| 662 | res_count_tuple) | ||
| 660 | 663 | ||
| 661 | # 3.创建月份表、提取/高亮关键行 | 664 | # 3.创建月份表、提取/高亮关键行 |
| 662 | # 倒序处理 | 665 | # 倒序处理 |
| ... | @@ -737,21 +740,35 @@ class BSWorkbook(Workbook): | ... | @@ -737,21 +740,35 @@ class BSWorkbook(Workbook): |
| 737 | res_list.sort(key=lambda x: (x[0], x[1], x[2])) | 740 | res_list.sort(key=lambda x: (x[0], x[1], x[2])) |
| 738 | ws = self.create_sheet(consts.RES_SHEET_NAME) | 741 | ws = self.create_sheet(consts.RES_SHEET_NAME) |
| 739 | ws.append(consts.RES_SHEET_HEADER) | 742 | ws.append(consts.RES_SHEET_HEADER) |
| 743 | success_count = 0 | ||
| 740 | for res_tuple in res_list: | 744 | for res_tuple in res_list: |
| 745 | if res_tuple[-1] not in consts.RES_FAILED_SET: | ||
| 746 | success_count += 1 | ||
| 741 | ws.append(res_tuple) | 747 | ws.append(res_tuple) |
| 748 | return len(res_list), success_count | ||
| 749 | else: | ||
| 750 | return 0, 0 | ||
| 751 | |||
| 752 | def move_res_sheet(self): | ||
| 753 | sheet = self.get_sheet_by_name(consts.RES_SHEET_NAME) | ||
| 754 | idx = self._sheets.index(sheet) | ||
| 755 | del self._sheets[idx] | ||
| 756 | self._sheets.append(sheet) | ||
| 742 | 757 | ||
| 743 | def remove_base_sheet(self): | 758 | def remove_base_sheet(self): |
| 744 | if len(self.sheetnames) > 1: | 759 | if len(self.sheetnames) > 1: |
| 745 | self.remove(self.get_sheet_by_name('Sheet')) | 760 | self.remove(self.get_sheet_by_name('Sheet')) |
| 746 | 761 | ||
| 747 | def rebuild(self, bs_summary, license_summary, res_list, document_scheme): | 762 | def rebuild(self, bs_summary, license_summary, res_list, document_scheme): |
| 763 | res_count_tuple = self.res_sheet(res_list) | ||
| 764 | |||
| 748 | count_list = [(consts.MODEL_FIELD_BS, len(bs_summary))] | 765 | count_list = [(consts.MODEL_FIELD_BS, len(bs_summary))] |
| 749 | if document_scheme == consts.DOC_SCHEME_LIST[1]: | 766 | if document_scheme == consts.DOC_SCHEME_LIST[1]: |
| 750 | self.license_rebuild(license_summary, document_scheme, count_list) | 767 | self.license_rebuild(license_summary, document_scheme, count_list) |
| 751 | self.bs_rebuild(bs_summary) | 768 | self.bs_rebuild(bs_summary, res_count_tuple) |
| 752 | else: | 769 | else: |
| 753 | self.bs_rebuild(bs_summary) | 770 | self.bs_rebuild(bs_summary, res_count_tuple) |
| 754 | self.license_rebuild(license_summary, document_scheme, count_list) | 771 | self.license_rebuild(license_summary, document_scheme, count_list) |
| 755 | self.res_sheet(res_list) | 772 | self.move_res_sheet() |
| 756 | self.remove_base_sheet() | 773 | self.remove_base_sheet() |
| 757 | return count_list | 774 | return count_list | ... | ... |
-
Please register or sign in to post a comment