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