diff --git a/code/paddle-ocr-demo/output/general_ocr_002_ocr_res_img.png b/code/paddle-ocr-demo/output/general_ocr_002_ocr_res_img.png index f3915b4d..324d915f 100644 Binary files a/code/paddle-ocr-demo/output/general_ocr_002_ocr_res_img.png and b/code/paddle-ocr-demo/output/general_ocr_002_ocr_res_img.png differ diff --git a/code/paddle-ocr-demo/output/general_ocr_002_preprocessed_img.png b/code/paddle-ocr-demo/output/general_ocr_002_preprocessed_img.png index 30de27d7..40222511 100644 Binary files a/code/paddle-ocr-demo/output/general_ocr_002_preprocessed_img.png and b/code/paddle-ocr-demo/output/general_ocr_002_preprocessed_img.png differ diff --git a/code/paddle-ocr-demo/output/general_ocr_002_res.json b/code/paddle-ocr-demo/output/general_ocr_002_res.json index 9f24599a..e0b3dfb0 100644 --- a/code/paddle-ocr-demo/output/general_ocr_002_res.json +++ b/code/paddle-ocr-demo/output/general_ocr_002_res.json @@ -1,5 +1,5 @@ { - "input_path": "C:/Users/Administrator/.paddlex/predict_input/general_ocr_002.png", + "input_path": "C:/Users/ONEAO/.paddlex/predict_input/general_ocr_002.png", "page_index": null, "model_settings": { "use_doc_preprocessor": true, @@ -691,39 +691,39 @@ "登机口于起飞前1O分钟关闭GATESCLOSE1OMINUTESBEFORE DEPARTURETIME" ], "rec_scores": [ - 0.9971157908439636, - 0.980294942855835, - 0.9757581949234009, - 0.9982836842536926, - 0.9992735981941223, - 0.9847624897956848, - 0.999160885810852, - 0.9843438267707825, - 0.9987466931343079, - 0.9556241631507874, - 0.9915273189544678, - 0.9323275089263916, - 0.795452356338501, - 0.9984054565429688, - 0.9987124800682068, - 0.9900163412094116, - 0.9853596687316895, - 0.9967120885848999, - 0.998457133769989, - 0.9965346455574036, - 0.9475476145744324, - 0.9983587265014648, - 0.9896446466445923, - 0.9970149397850037, - 0.9870915412902832, - 0.9862422943115234, - 0.9981896281242371, - 0.988991379737854, - 0.9749718308448792, - 0.9971947073936462, - 0.9975488781929016, - 0.9928285479545593, - 0.9732762575149536 + 0.9971110224723816, + 0.9802942276000977, + 0.9757574796676636, + 0.9982827305793762, + 0.999271810054779, + 0.9847618937492371, + 0.9991598129272461, + 0.984343409538269, + 0.9987451434135437, + 0.9556233286857605, + 0.9915266633033752, + 0.932326078414917, + 0.7954467535018921, + 0.9984052777290344, + 0.9987090229988098, + 0.9900156855583191, + 0.9853563904762268, + 0.996711790561676, + 0.9984555244445801, + 0.9965342879295349, + 0.9475457072257996, + 0.9983549118041992, + 0.9896441102027893, + 0.9970144629478455, + 0.9870910048484802, + 0.986241340637207, + 0.9981892108917236, + 0.9889904856681824, + 0.974971354007721, + 0.9971852898597717, + 0.997547447681427, + 0.9928280711174011, + 0.9732754230499268 ], "rec_polys": [ [ diff --git a/code/paddle-ocr-demo/output/test1_ocr_res_img.png b/code/paddle-ocr-demo/output/test1_ocr_res_img.png index c9dc5c7b..6601cd97 100644 Binary files a/code/paddle-ocr-demo/output/test1_ocr_res_img.png and b/code/paddle-ocr-demo/output/test1_ocr_res_img.png differ diff --git a/code/paddle-ocr-demo/output/test1_preprocessed_img.png b/code/paddle-ocr-demo/output/test1_preprocessed_img.png index 3258b6a9..f7b9569e 100644 Binary files a/code/paddle-ocr-demo/output/test1_preprocessed_img.png and b/code/paddle-ocr-demo/output/test1_preprocessed_img.png differ diff --git a/code/paddle-ocr-demo/output/test1_res.json b/code/paddle-ocr-demo/output/test1_res.json index 0ce17185..0d0e79cf 100644 --- a/code/paddle-ocr-demo/output/test1_res.json +++ b/code/paddle-ocr-demo/output/test1_res.json @@ -1271,68 +1271,68 @@ "fOrPACIFICINTERNATIONALLINES(PTE)LTDaSCarrier" ], "rec_scores": [ - 0.9938989281654358, - 0.972846508026123, - 0.9881200194358826, - 0.990138828754425, - 0.9878993034362793, - 0.9665594696998596, - 0.9608297348022461, - 0.9596557021141052, - 0.9911079406738281, - 0.9478586316108704, - 0.9497116208076477, - 0.9310017228126526, - 0.9194626212120056, - 0.9266672730445862, - 0.9126367568969727, - 0.8457321524620056, - 0.9567287564277649, - 0.9139136672019958, - 0.9166555404663086, - 0.9256715774536133, - 0.9291625618934631, - 0.9593757390975952, - 0.9525982737541199, - 0.954907238483429, - 0.9660497307777405, - 0.9483555555343628, - 0.9553540349006653, - 0.9510287642478943, - 0.9745871424674988, - 0.9968070387840271, - 0.9835952520370483, - 0.9761207699775696, - 0.9424749612808228, - 0.9949460625648499, - 0.93708336353302, - 0.964686393737793, - 0.9537389278411865, - 0.9928998351097107, - 0.9603193998336792, - 0.9215961694717407, - 0.9375246167182922, - 0.9979581236839294, - 0.9746990203857422, - 0.9987379908561707, - 0.9981704950332642, - 0.9976575970649719, - 0.9971591830253601, - 0.9966489672660828, - 0.9627765417098999, - 0.9630685448646545, - 0.9711615443229675, - 0.9970062375068665, - 0.9598766565322876, - 0.9645793437957764, - 0.9334232211112976, - 0.9843224287033081, - 0.9368222951889038, - 0.9272780418395996, - 0.9519227743148804, - 0.9674265384674072, - 0.957920253276825, - 0.9185543656349182 + 0.993898332118988, + 0.9728460311889648, + 0.9881201982498169, + 0.9901384711265564, + 0.9878991842269897, + 0.9665590524673462, + 0.960828423500061, + 0.9596601128578186, + 0.9911072850227356, + 0.9478574991226196, + 0.9497101902961731, + 0.9310002326965332, + 0.9194604754447937, + 0.9266656041145325, + 0.9126342535018921, + 0.8457291722297668, + 0.9567282795906067, + 0.9139121770858765, + 0.9166549444198608, + 0.9256700873374939, + 0.9291608333587646, + 0.959374725818634, + 0.9525966644287109, + 0.9549065232276917, + 0.9660487771034241, + 0.9483537673950195, + 0.955353319644928, + 0.951028048992157, + 0.9745868444442749, + 0.9968065023422241, + 0.9835948348045349, + 0.9761202335357666, + 0.9424735307693481, + 0.9949454069137573, + 0.9370826482772827, + 0.9646853804588318, + 0.9537377953529358, + 0.9928991794586182, + 0.9603179693222046, + 0.9215952754020691, + 0.9375240206718445, + 0.9979577660560608, + 0.9746984839439392, + 0.9987377524375916, + 0.9981698989868164, + 0.9976568818092346, + 0.9971590042114258, + 0.996648907661438, + 0.9627759456634521, + 0.9630667567253113, + 0.9711611866950989, + 0.9970054030418396, + 0.9598758220672607, + 0.9645780920982361, + 0.9334222674369812, + 0.9843217730522156, + 0.9368213415145874, + 0.927276611328125, + 0.9519212245941162, + 0.9674260020256042, + 0.9579195976257324, + 0.9185529947280884 ], "rec_polys": [ [ diff --git a/code/paddle-ocr-demo/output/time_ocr_res_img.jpg b/code/paddle-ocr-demo/output/time_ocr_res_img.jpg deleted file mode 100644 index a1431015..00000000 Binary files a/code/paddle-ocr-demo/output/time_ocr_res_img.jpg and /dev/null differ diff --git a/code/paddle-ocr-demo/output/time_preprocessed_img.jpg b/code/paddle-ocr-demo/output/time_preprocessed_img.jpg deleted file mode 100644 index 962f6d26..00000000 Binary files a/code/paddle-ocr-demo/output/time_preprocessed_img.jpg and /dev/null differ diff --git a/code/paddle-ocr-demo/output/time_res.json b/code/paddle-ocr-demo/output/time_res.json deleted file mode 100644 index c311277a..00000000 --- a/code/paddle-ocr-demo/output/time_res.json +++ /dev/null @@ -1,3099 +0,0 @@ -{ - "input_path": "time.jpg", - "page_index": null, - "model_settings": { - "use_doc_preprocessor": true, - "use_textline_orientation": false - }, - "doc_preprocessor_res": { - "input_path": null, - "page_index": null, - "model_settings": { - "use_doc_orientation_classify": false, - "use_doc_unwarping": false - }, - "angle": -1 - }, - "dt_polys": [ - [ - [ - 1, - 2 - ], - [ - 84, - 2 - ], - [ - 84, - 23 - ], - [ - 1, - 23 - ] - ], - [ - [ - 92, - 2 - ], - [ - 171, - 2 - ], - [ - 171, - 23 - ], - [ - 92, - 23 - ] - ], - [ - [ - 249, - 31 - ], - [ - 376, - 31 - ], - [ - 376, - 47 - ], - [ - 249, - 47 - ] - ], - [ - [ - 163, - 51 - ], - [ - 274, - 51 - ], - [ - 274, - 71 - ], - [ - 163, - 71 - ] - ], - [ - [ - 496, - 52 - ], - [ - 568, - 52 - ], - [ - 568, - 70 - ], - [ - 496, - 70 - ] - ], - [ - [ - 35, - 76 - ], - [ - 399, - 76 - ], - [ - 399, - 93 - ], - [ - 35, - 93 - ] - ], - [ - [ - 37, - 102 - ], - [ - 398, - 102 - ], - [ - 398, - 116 - ], - [ - 37, - 116 - ] - ], - [ - [ - 35, - 122 - ], - [ - 209, - 122 - ], - [ - 209, - 141 - ], - [ - 35, - 141 - ] - ], - [ - [ - 503, - 122 - ], - [ - 567, - 122 - ], - [ - 567, - 139 - ], - [ - 503, - 139 - ] - ], - [ - [ - 33, - 143 - ], - [ - 208, - 143 - ], - [ - 208, - 163 - ], - [ - 33, - 163 - ] - ], - [ - [ - 425, - 142 - ], - [ - 653, - 142 - ], - [ - 653, - 163 - ], - [ - 425, - 163 - ] - ], - [ - [ - 157, - 169 - ], - [ - 281, - 169 - ], - [ - 281, - 189 - ], - [ - 157, - 189 - ] - ], - [ - [ - 41, - 194 - ], - [ - 289, - 194 - ], - [ - 289, - 211 - ], - [ - 41, - 211 - ] - ], - [ - [ - 416, - 194 - ], - [ - 653, - 194 - ], - [ - 653, - 211 - ], - [ - 416, - 211 - ] - ], - [ - [ - 36, - 218 - ], - [ - 220, - 218 - ], - [ - 220, - 234 - ], - [ - 36, - 234 - ] - ], - [ - [ - 251, - 216 - ], - [ - 307, - 216 - ], - [ - 307, - 236 - ], - [ - 251, - 236 - ] - ], - [ - [ - 338, - 216 - ], - [ - 404, - 216 - ], - [ - 404, - 236 - ], - [ - 338, - 236 - ] - ], - [ - [ - 33, - 239 - ], - [ - 151, - 239 - ], - [ - 151, - 259 - ], - [ - 33, - 259 - ] - ], - [ - [ - 575, - 237 - ], - [ - 640, - 237 - ], - [ - 640, - 258 - ], - [ - 575, - 258 - ] - ], - [ - [ - 147, - 264 - ], - [ - 289, - 264 - ], - [ - 289, - 281 - ], - [ - 147, - 281 - ] - ], - [ - [ - 34, - 287 - ], - [ - 283, - 288 - ], - [ - 283, - 306 - ], - [ - 34, - 305 - ] - ], - [ - [ - 440, - 287 - ], - [ - 626, - 287 - ], - [ - 626, - 304 - ], - [ - 440, - 304 - ] - ], - [ - [ - 35, - 311 - ], - [ - 218, - 311 - ], - [ - 218, - 327 - ], - [ - 35, - 327 - ] - ], - [ - [ - 251, - 310 - ], - [ - 307, - 310 - ], - [ - 307, - 330 - ], - [ - 251, - 330 - ] - ], - [ - [ - 338, - 310 - ], - [ - 404, - 310 - ], - [ - 404, - 330 - ], - [ - 338, - 330 - ] - ], - [ - [ - 36, - 335 - ], - [ - 151, - 335 - ], - [ - 151, - 352 - ], - [ - 36, - 352 - ] - ], - [ - [ - 39, - 359 - ], - [ - 208, - 359 - ], - [ - 208, - 376 - ], - [ - 39, - 376 - ] - ], - [ - [ - 223, - 359 - ], - [ - 352, - 359 - ], - [ - 352, - 376 - ], - [ - 223, - 376 - ] - ], - [ - [ - 46, - 407 - ], - [ - 201, - 407 - ], - [ - 201, - 423 - ], - [ - 46, - 423 - ] - ], - [ - [ - 225, - 407 - ], - [ - 387, - 407 - ], - [ - 387, - 423 - ], - [ - 225, - 423 - ] - ], - [ - [ - 276, - 426 - ], - [ - 340, - 426 - ], - [ - 340, - 451 - ], - [ - 276, - 451 - ] - ], - [ - [ - 35, - 454 - ], - [ - 215, - 454 - ], - [ - 215, - 471 - ], - [ - 35, - 471 - ] - ], - [ - [ - 220, - 453 - ], - [ - 400, - 453 - ], - [ - 400, - 473 - ], - [ - 220, - 473 - ] - ], - [ - [ - 280, - 476 - ], - [ - 346, - 476 - ], - [ - 346, - 498 - ], - [ - 280, - 498 - ] - ], - [ - [ - 52, - 501 - ], - [ - 637, - 501 - ], - [ - 637, - 520 - ], - [ - 52, - 520 - ] - ], - [ - [ - 215, - 522 - ], - [ - 471, - 522 - ], - [ - 471, - 542 - ], - [ - 215, - 542 - ] - ], - [ - [ - 30, - 547 - ], - [ - 65, - 547 - ], - [ - 65, - 571 - ], - [ - 30, - 571 - ] - ], - [ - [ - 193, - 549 - ], - [ - 376, - 549 - ], - [ - 376, - 569 - ], - [ - 193, - 569 - ] - ], - [ - [ - 494, - 547 - ], - [ - 537, - 547 - ], - [ - 537, - 570 - ], - [ - 494, - 570 - ] - ], - [ - [ - 540, - 547 - ], - [ - 577, - 547 - ], - [ - 577, - 570 - ], - [ - 540, - 570 - ] - ], - [ - [ - 589, - 547 - ], - [ - 658, - 547 - ], - [ - 658, - 569 - ], - [ - 589, - 569 - ] - ], - [ - [ - 378, - 571 - ], - [ - 461, - 571 - ], - [ - 461, - 592 - ], - [ - 378, - 592 - ] - ], - [ - [ - 377, - 596 - ], - [ - 451, - 593 - ], - [ - 452, - 614 - ], - [ - 378, - 617 - ] - ], - [ - [ - 114, - 623 - ], - [ - 576, - 620 - ], - [ - 576, - 641 - ], - [ - 114, - 644 - ] - ], - [ - [ - 46, - 679 - ], - [ - 166, - 679 - ], - [ - 166, - 695 - ], - [ - 46, - 695 - ] - ], - [ - [ - 192, - 679 - ], - [ - 281, - 679 - ], - [ - 281, - 696 - ], - [ - 192, - 696 - ] - ], - [ - [ - 320, - 679 - ], - [ - 356, - 679 - ], - [ - 356, - 698 - ], - [ - 320, - 698 - ] - ], - [ - [ - 427, - 676 - ], - [ - 459, - 676 - ], - [ - 459, - 699 - ], - [ - 427, - 699 - ] - ], - [ - [ - 498, - 679 - ], - [ - 551, - 679 - ], - [ - 551, - 696 - ], - [ - 498, - 696 - ] - ], - [ - [ - 574, - 678 - ], - [ - 632, - 678 - ], - [ - 632, - 698 - ], - [ - 574, - 698 - ] - ], - [ - [ - 69, - 699 - ], - [ - 147, - 699 - ], - [ - 147, - 724 - ], - [ - 69, - 724 - ] - ], - [ - [ - 193, - 701 - ], - [ - 261, - 701 - ], - [ - 261, - 723 - ], - [ - 193, - 723 - ] - ], - [ - [ - 320, - 698 - ], - [ - 363, - 698 - ], - [ - 363, - 727 - ], - [ - 320, - 727 - ] - ], - [ - [ - 497, - 699 - ], - [ - 539, - 699 - ], - [ - 539, - 727 - ], - [ - 497, - 727 - ] - ], - [ - [ - 575, - 701 - ], - [ - 612, - 701 - ], - [ - 612, - 724 - ], - [ - 575, - 724 - ] - ], - [ - [ - 54, - 727 - ], - [ - 160, - 727 - ], - [ - 160, - 744 - ], - [ - 54, - 744 - ] - ], - [ - [ - 198, - 727 - ], - [ - 266, - 727 - ], - [ - 266, - 745 - ], - [ - 198, - 745 - ] - ], - [ - [ - 53, - 774 - ], - [ - 161, - 774 - ], - [ - 161, - 793 - ], - [ - 53, - 793 - ] - ], - [ - [ - 192, - 776 - ], - [ - 298, - 776 - ], - [ - 298, - 792 - ], - [ - 192, - 792 - ] - ], - [ - [ - 433, - 777 - ], - [ - 609, - 777 - ], - [ - 609, - 790 - ], - [ - 433, - 790 - ] - ], - [ - [ - 441, - 798 - ], - [ - 635, - 798 - ], - [ - 635, - 815 - ], - [ - 441, - 815 - ] - ], - [ - [ - 62, - 870 - ], - [ - 153, - 870 - ], - [ - 153, - 886 - ], - [ - 62, - 886 - ] - ], - [ - [ - 192, - 870 - ], - [ - 378, - 870 - ], - [ - 378, - 886 - ], - [ - 192, - 886 - ] - ], - [ - [ - 77, - 892 - ], - [ - 138, - 889 - ], - [ - 139, - 911 - ], - [ - 78, - 913 - ] - ], - [ - [ - 192, - 892 - ], - [ - 305, - 890 - ], - [ - 305, - 911 - ], - [ - 192, - 913 - ] - ], - [ - [ - 133, - 918 - ], - [ - 319, - 918 - ], - [ - 319, - 935 - ], - [ - 133, - 935 - ] - ], - [ - [ - 430, - 917 - ], - [ - 465, - 917 - ], - [ - 465, - 937 - ], - [ - 430, - 937 - ] - ], - [ - [ - 198, - 941 - ], - [ - 255, - 941 - ], - [ - 255, - 959 - ], - [ - 198, - 959 - ] - ] - ], - "text_det_params": { - "limit_side_len": 64, - "limit_type": "min", - "thresh": 0.3, - "max_side_limit": 4000, - "box_thresh": 0.6, - "unclip_ratio": 1.5 - }, - "text_type": "general", - "textline_orientation_angles": [ - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1, - -1 - ], - "text_rec_score_thresh": 0.0, - "rec_texts": [ - "式样9-11", - "海运提单", - "BILLOFLADING", - "1)SHIPPER托运人", - "10)B/LNO.", - "ZE GUO AIR COMPRESSOR INDUSTRIAL ZONE", - "WENLINGCITY.ZHEJIANGPROVINCE,CHINA", - "TEL:86-576-86402201", - "COSCO", - "FAX:86-576-86402198", - "中国远洋运输（集团）总公司", - "2)CONSIGNEE收货人", - "PT.BADJAABADISENTOSAPERTOKOAN", - "CHINA OCEANSHIPPING(GROUP)CO.", - "GLODOKJAYNO90JL.HAYAM", - "WURUK", - "JAKARTA", - "11180,INDONESIA", - "ORIGINAL", - "3）NOTIFYPARTY通知人", - "PT.BADJAABADISENTOSAPERTOKOAN", - "CONBINED TRANPORT BILL OF LADIIE", - "GLODOKJAYNO.90JL.HAYAM", - "WURUK", - "JAKARTA", - "11180.INDONESIA", - "4PLACEOFRECEIPT收货地", - "5）OCEAN VESSEL船名", - "6）VOYAGENO.船名航次", - "7PORTOFLOADING装运港", - "Ningbo", - "8PORTOF DISCHARGE卸货港", - "9）PLACE OF DELIVERY目的港", - "JAKARTA", - "11)MARKS唛头12)NOS.&KINDS OFPKGS包装与件数13）DESCRIPTIONOFGOODS商品名称", - "14)G.W.(kg）毛重和体积15)MEAS(m3)", - "N/M", - "184WOODEN CASES 1*20HQ", - "CUT", - "OFF", - "MACHINE", - "9660.0KGS", - "30.734CBM", - "16)TOTAL NUMBEROFCONTAINERSOR PACKAGES(INWORDS)总件数184PCS", - "FREIGHT & CHARGES", - "REVENUE TONS", - "RATE", - "PER", - "PREPAID", - "COLLECT", - "运费支付", - "计费吨数", - "价格", - "预付", - "到付", - "FREIGHT PREPAID", - "9660.0KGS", - "PREPAIDAT预付", - "PAYABLEAT应付", - "17)PLACEANDDATEOFISSUE", - "签发地点与日期NOV292011", - "TOTAL PREPAID", - "18）NUMBEROFORIGINALB（S)L", - "总额预付", - "正本提单份数3", - "LOADING ON BOARD THE VESSEL", - "20)B", - "19)DATE" - ], - "rec_scores": [ - 0.9968595504760742, - 0.9998053312301636, - 0.9978482127189636, - 0.9521610140800476, - 0.9378439784049988, - 0.9624345302581787, - 0.9586712121963501, - 0.9875702261924744, - 0.8193227052688599, - 0.9899468421936035, - 0.9970354437828064, - 0.9355073571205139, - 0.9913122057914734, - 0.981356143951416, - 0.9539093971252441, - 0.9930180311203003, - 0.9981914758682251, - 0.968727171421051, - 0.9957302808761597, - 0.9597290754318237, - 0.9729806184768677, - 0.877500057220459, - 0.9232662320137024, - 0.9934558868408203, - 0.9971846342086792, - 0.9407088756561279, - 0.9934617280960083, - 0.9322524070739746, - 0.9447787404060364, - 0.9590057134628296, - 0.998225212097168, - 0.9751702547073364, - 0.9334163069725037, - 0.9984990358352661, - 0.9399874210357666, - 0.8810221552848816, - 0.9610098004341125, - 0.9265491366386414, - 0.9792051315307617, - 0.9824749827384949, - 0.9977664947509766, - 0.9980397820472717, - 0.9939849376678467, - 0.9261623024940491, - 0.9516990780830383, - 0.9644684791564941, - 0.9965518712997437, - 0.9918086528778076, - 0.9971172213554382, - 0.9963092803955078, - 0.9998077154159546, - 0.999522864818573, - 0.9997318983078003, - 0.9995837211608887, - 0.9994396567344666, - 0.9807721376419067, - 0.9829018115997314, - 0.9972453713417053, - 0.9892377853393555, - 0.9400802254676819, - 0.9922091960906982, - 0.9825478196144104, - 0.9257330894470215, - 0.9994402527809143, - 0.9995817542076111, - 0.9415965676307678, - 0.9494259357452393, - 0.9667239189147949 - ], - "rec_polys": [ - [ - [ - 1, - 2 - ], - [ - 84, - 2 - ], - [ - 84, - 23 - ], - [ - 1, - 23 - ] - ], - [ - [ - 92, - 2 - ], - [ - 171, - 2 - ], - [ - 171, - 23 - ], - [ - 92, - 23 - ] - ], - [ - [ - 249, - 31 - ], - [ - 376, - 31 - ], - [ - 376, - 47 - ], - [ - 249, - 47 - ] - ], - [ - [ - 163, - 51 - ], - [ - 274, - 51 - ], - [ - 274, - 71 - ], - [ - 163, - 71 - ] - ], - [ - [ - 496, - 52 - ], - [ - 568, - 52 - ], - [ - 568, - 70 - ], - [ - 496, - 70 - ] - ], - [ - [ - 35, - 76 - ], - [ - 399, - 76 - ], - [ - 399, - 93 - ], - [ - 35, - 93 - ] - ], - [ - [ - 37, - 102 - ], - [ - 398, - 102 - ], - [ - 398, - 116 - ], - [ - 37, - 116 - ] - ], - [ - [ - 35, - 122 - ], - [ - 209, - 122 - ], - [ - 209, - 141 - ], - [ - 35, - 141 - ] - ], - [ - [ - 503, - 122 - ], - [ - 567, - 122 - ], - [ - 567, - 139 - ], - [ - 503, - 139 - ] - ], - [ - [ - 33, - 143 - ], - [ - 208, - 143 - ], - [ - 208, - 163 - ], - [ - 33, - 163 - ] - ], - [ - [ - 425, - 142 - ], - [ - 653, - 142 - ], - [ - 653, - 163 - ], - [ - 425, - 163 - ] - ], - [ - [ - 157, - 169 - ], - [ - 281, - 169 - ], - [ - 281, - 189 - ], - [ - 157, - 189 - ] - ], - [ - [ - 41, - 194 - ], - [ - 289, - 194 - ], - [ - 289, - 211 - ], - [ - 41, - 211 - ] - ], - [ - [ - 416, - 194 - ], - [ - 653, - 194 - ], - [ - 653, - 211 - ], - [ - 416, - 211 - ] - ], - [ - [ - 36, - 218 - ], - [ - 220, - 218 - ], - [ - 220, - 234 - ], - [ - 36, - 234 - ] - ], - [ - [ - 251, - 216 - ], - [ - 307, - 216 - ], - [ - 307, - 236 - ], - [ - 251, - 236 - ] - ], - [ - [ - 338, - 216 - ], - [ - 404, - 216 - ], - [ - 404, - 236 - ], - [ - 338, - 236 - ] - ], - [ - [ - 33, - 239 - ], - [ - 151, - 239 - ], - [ - 151, - 259 - ], - [ - 33, - 259 - ] - ], - [ - [ - 575, - 237 - ], - [ - 640, - 237 - ], - [ - 640, - 258 - ], - [ - 575, - 258 - ] - ], - [ - [ - 147, - 264 - ], - [ - 289, - 264 - ], - [ - 289, - 281 - ], - [ - 147, - 281 - ] - ], - [ - [ - 34, - 287 - ], - [ - 283, - 288 - ], - [ - 283, - 306 - ], - [ - 34, - 305 - ] - ], - [ - [ - 440, - 287 - ], - [ - 626, - 287 - ], - [ - 626, - 304 - ], - [ - 440, - 304 - ] - ], - [ - [ - 35, - 311 - ], - [ - 218, - 311 - ], - [ - 218, - 327 - ], - [ - 35, - 327 - ] - ], - [ - [ - 251, - 310 - ], - [ - 307, - 310 - ], - [ - 307, - 330 - ], - [ - 251, - 330 - ] - ], - [ - [ - 338, - 310 - ], - [ - 404, - 310 - ], - [ - 404, - 330 - ], - [ - 338, - 330 - ] - ], - [ - [ - 36, - 335 - ], - [ - 151, - 335 - ], - [ - 151, - 352 - ], - [ - 36, - 352 - ] - ], - [ - [ - 39, - 359 - ], - [ - 208, - 359 - ], - [ - 208, - 376 - ], - [ - 39, - 376 - ] - ], - [ - [ - 223, - 359 - ], - [ - 352, - 359 - ], - [ - 352, - 376 - ], - [ - 223, - 376 - ] - ], - [ - [ - 46, - 407 - ], - [ - 201, - 407 - ], - [ - 201, - 423 - ], - [ - 46, - 423 - ] - ], - [ - [ - 225, - 407 - ], - [ - 387, - 407 - ], - [ - 387, - 423 - ], - [ - 225, - 423 - ] - ], - [ - [ - 276, - 426 - ], - [ - 340, - 426 - ], - [ - 340, - 451 - ], - [ - 276, - 451 - ] - ], - [ - [ - 35, - 454 - ], - [ - 215, - 454 - ], - [ - 215, - 471 - ], - [ - 35, - 471 - ] - ], - [ - [ - 220, - 453 - ], - [ - 400, - 453 - ], - [ - 400, - 473 - ], - [ - 220, - 473 - ] - ], - [ - [ - 280, - 476 - ], - [ - 346, - 476 - ], - [ - 346, - 498 - ], - [ - 280, - 498 - ] - ], - [ - [ - 52, - 501 - ], - [ - 637, - 501 - ], - [ - 637, - 520 - ], - [ - 52, - 520 - ] - ], - [ - [ - 215, - 522 - ], - [ - 471, - 522 - ], - [ - 471, - 542 - ], - [ - 215, - 542 - ] - ], - [ - [ - 30, - 547 - ], - [ - 65, - 547 - ], - [ - 65, - 571 - ], - [ - 30, - 571 - ] - ], - [ - [ - 193, - 549 - ], - [ - 376, - 549 - ], - [ - 376, - 569 - ], - [ - 193, - 569 - ] - ], - [ - [ - 494, - 547 - ], - [ - 537, - 547 - ], - [ - 537, - 570 - ], - [ - 494, - 570 - ] - ], - [ - [ - 540, - 547 - ], - [ - 577, - 547 - ], - [ - 577, - 570 - ], - [ - 540, - 570 - ] - ], - [ - [ - 589, - 547 - ], - [ - 658, - 547 - ], - [ - 658, - 569 - ], - [ - 589, - 569 - ] - ], - [ - [ - 378, - 571 - ], - [ - 461, - 571 - ], - [ - 461, - 592 - ], - [ - 378, - 592 - ] - ], - [ - [ - 377, - 596 - ], - [ - 451, - 593 - ], - [ - 452, - 614 - ], - [ - 378, - 617 - ] - ], - [ - [ - 114, - 623 - ], - [ - 576, - 620 - ], - [ - 576, - 641 - ], - [ - 114, - 644 - ] - ], - [ - [ - 46, - 679 - ], - [ - 166, - 679 - ], - [ - 166, - 695 - ], - [ - 46, - 695 - ] - ], - [ - [ - 192, - 679 - ], - [ - 281, - 679 - ], - [ - 281, - 696 - ], - [ - 192, - 696 - ] - ], - [ - [ - 320, - 679 - ], - [ - 356, - 679 - ], - [ - 356, - 698 - ], - [ - 320, - 698 - ] - ], - [ - [ - 427, - 676 - ], - [ - 459, - 676 - ], - [ - 459, - 699 - ], - [ - 427, - 699 - ] - ], - [ - [ - 498, - 679 - ], - [ - 551, - 679 - ], - [ - 551, - 696 - ], - [ - 498, - 696 - ] - ], - [ - [ - 574, - 678 - ], - [ - 632, - 678 - ], - [ - 632, - 698 - ], - [ - 574, - 698 - ] - ], - [ - [ - 69, - 699 - ], - [ - 147, - 699 - ], - [ - 147, - 724 - ], - [ - 69, - 724 - ] - ], - [ - [ - 193, - 701 - ], - [ - 261, - 701 - ], - [ - 261, - 723 - ], - [ - 193, - 723 - ] - ], - [ - [ - 320, - 698 - ], - [ - 363, - 698 - ], - [ - 363, - 727 - ], - [ - 320, - 727 - ] - ], - [ - [ - 497, - 699 - ], - [ - 539, - 699 - ], - [ - 539, - 727 - ], - [ - 497, - 727 - ] - ], - [ - [ - 575, - 701 - ], - [ - 612, - 701 - ], - [ - 612, - 724 - ], - [ - 575, - 724 - ] - ], - [ - [ - 54, - 727 - ], - [ - 160, - 727 - ], - [ - 160, - 744 - ], - [ - 54, - 744 - ] - ], - [ - [ - 198, - 727 - ], - [ - 266, - 727 - ], - [ - 266, - 745 - ], - [ - 198, - 745 - ] - ], - [ - [ - 53, - 774 - ], - [ - 161, - 774 - ], - [ - 161, - 793 - ], - [ - 53, - 793 - ] - ], - [ - [ - 192, - 776 - ], - [ - 298, - 776 - ], - [ - 298, - 792 - ], - [ - 192, - 792 - ] - ], - [ - [ - 433, - 777 - ], - [ - 609, - 777 - ], - [ - 609, - 790 - ], - [ - 433, - 790 - ] - ], - [ - [ - 441, - 798 - ], - [ - 635, - 798 - ], - [ - 635, - 815 - ], - [ - 441, - 815 - ] - ], - [ - [ - 62, - 870 - ], - [ - 153, - 870 - ], - [ - 153, - 886 - ], - [ - 62, - 886 - ] - ], - [ - [ - 192, - 870 - ], - [ - 378, - 870 - ], - [ - 378, - 886 - ], - [ - 192, - 886 - ] - ], - [ - [ - 77, - 892 - ], - [ - 138, - 889 - ], - [ - 139, - 911 - ], - [ - 78, - 913 - ] - ], - [ - [ - 192, - 892 - ], - [ - 305, - 890 - ], - [ - 305, - 911 - ], - [ - 192, - 913 - ] - ], - [ - [ - 133, - 918 - ], - [ - 319, - 918 - ], - [ - 319, - 935 - ], - [ - 133, - 935 - ] - ], - [ - [ - 430, - 917 - ], - [ - 465, - 917 - ], - [ - 465, - 937 - ], - [ - 430, - 937 - ] - ], - [ - [ - 198, - 941 - ], - [ - 255, - 941 - ], - [ - 255, - 959 - ], - [ - 198, - 959 - ] - ] - ], - "rec_boxes": [ - [ - 1, - 2, - 84, - 23 - ], - [ - 92, - 2, - 171, - 23 - ], - [ - 249, - 31, - 376, - 47 - ], - [ - 163, - 51, - 274, - 71 - ], - [ - 496, - 52, - 568, - 70 - ], - [ - 35, - 76, - 399, - 93 - ], - [ - 37, - 102, - 398, - 116 - ], - [ - 35, - 122, - 209, - 141 - ], - [ - 503, - 122, - 567, - 139 - ], - [ - 33, - 143, - 208, - 163 - ], - [ - 425, - 142, - 653, - 163 - ], - [ - 157, - 169, - 281, - 189 - ], - [ - 41, - 194, - 289, - 211 - ], - [ - 416, - 194, - 653, - 211 - ], - [ - 36, - 218, - 220, - 234 - ], - [ - 251, - 216, - 307, - 236 - ], - [ - 338, - 216, - 404, - 236 - ], - [ - 33, - 239, - 151, - 259 - ], - [ - 575, - 237, - 640, - 258 - ], - [ - 147, - 264, - 289, - 281 - ], - [ - 34, - 287, - 283, - 306 - ], - [ - 440, - 287, - 626, - 304 - ], - [ - 35, - 311, - 218, - 327 - ], - [ - 251, - 310, - 307, - 330 - ], - [ - 338, - 310, - 404, - 330 - ], - [ - 36, - 335, - 151, - 352 - ], - [ - 39, - 359, - 208, - 376 - ], - [ - 223, - 359, - 352, - 376 - ], - [ - 46, - 407, - 201, - 423 - ], - [ - 225, - 407, - 387, - 423 - ], - [ - 276, - 426, - 340, - 451 - ], - [ - 35, - 454, - 215, - 471 - ], - [ - 220, - 453, - 400, - 473 - ], - [ - 280, - 476, - 346, - 498 - ], - [ - 52, - 501, - 637, - 520 - ], - [ - 215, - 522, - 471, - 542 - ], - [ - 30, - 547, - 65, - 571 - ], - [ - 193, - 549, - 376, - 569 - ], - [ - 494, - 547, - 537, - 570 - ], - [ - 540, - 547, - 577, - 570 - ], - [ - 589, - 547, - 658, - 569 - ], - [ - 378, - 571, - 461, - 592 - ], - [ - 377, - 593, - 452, - 617 - ], - [ - 114, - 620, - 576, - 644 - ], - [ - 46, - 679, - 166, - 695 - ], - [ - 192, - 679, - 281, - 696 - ], - [ - 320, - 679, - 356, - 698 - ], - [ - 427, - 676, - 459, - 699 - ], - [ - 498, - 679, - 551, - 696 - ], - [ - 574, - 678, - 632, - 698 - ], - [ - 69, - 699, - 147, - 724 - ], - [ - 193, - 701, - 261, - 723 - ], - [ - 320, - 698, - 363, - 727 - ], - [ - 497, - 699, - 539, - 727 - ], - [ - 575, - 701, - 612, - 724 - ], - [ - 54, - 727, - 160, - 744 - ], - [ - 198, - 727, - 266, - 745 - ], - [ - 53, - 774, - 161, - 793 - ], - [ - 192, - 776, - 298, - 792 - ], - [ - 433, - 777, - 609, - 790 - ], - [ - 441, - 798, - 635, - 815 - ], - [ - 62, - 870, - 153, - 886 - ], - [ - 192, - 870, - 378, - 886 - ], - [ - 77, - 889, - 139, - 913 - ], - [ - 192, - 890, - 305, - 913 - ], - [ - 133, - 918, - 319, - 935 - ], - [ - 430, - 917, - 465, - 937 - ], - [ - 198, - 941, - 255, - 959 - ] - ] -} \ No newline at end of file diff --git a/code/paddle-ocr-demo/test.py b/code/paddle-ocr-demo/test.py index 401e5719..abf39ae2 100644 --- a/code/paddle-ocr-demo/test.py +++ b/code/paddle-ocr-demo/test.py @@ -1,21 +1,12 @@ -# from paddleocr import PaddleOCR -# # 初始化 PaddleOCR 实例 -# ocr = PaddleOCR( -# use_doc_orientation_classify=False, -# use_doc_unwarping=False, -# use_textline_orientation=False) -# # 对示例图像执行 OCR 推理 -# result = ocr.predict( -# input="test1.png") -# # 可视化结果并保存 json 结果 -# for res in result: -# res.print() -# res.save_to_img("output") -# res.save_to_json("output") - - - from paddleocr import PPChatOCRv4Doc +import paddle +# 检查 PaddlePaddle 是否安装成功 +# 打印是否支持 GPU +print("是否支持 GPU:", paddle.is_compiled_with_cuda()) + +# 如果支持，再看看当前是否在使用 GPU +if paddle.is_compiled_with_cuda(): + print("当前是否使用 GPU:", paddle.device.get_device()) chat_bot_config = { "module_name": "chat_bot", @@ -59,7 +50,7 @@ if use_mllm: mllm_predict_res = pipeline.mllm_pred( input="test1.png", - key_list=["Gross Weight"], + key_list=["Shipper"], mllm_chat_bot_config=mllm_chat_bot_config, ) mllm_predict_info = mllm_predict_res["mllm_res"] @@ -73,7 +64,7 @@ vector_info = pipeline.build_vector( visual_info_list, flag_save_bytes_vector=True, retriever_config=retriever_config ) chat_result = pipeline.chat( - key_list=["Gross Weight"], + key_list=["Shipper"], visual_info=visual_info_list, vector_info=vector_info, mllm_predict_info=mllm_predict_info,