Tong Gao ad470e323a
[Feature] Refactor Inferencers (#1608)
* tmp commit

* initial

* kie

* update MMOCRInferencer and ocr.py

* fix

* fix bug & add ut

* ut for kie

* part of mmocr inferencer ut

* part of mmocr inferencer ut

* ut

* ut

* docs

* inferencer

* Add TextSpotInferencer

* test

* fix

* textspot

* fix

* test

* test

* fix

* fix
2023-02-02 19:05:55 +08:00

3 lines
8.6 KiB
Plaintext

{"file_name": "tests/data/kie_toy_dataset/wildreceipt/1.jpeg", "height": 1200, "width": 1600, "annotations": [{"box": [550.0, 190.0, 937.0, 190.0, 937.0, 104.0, 550.0, 104.0], "text": "SAFEWAY", "label": 1}, {"box": [1048.0, 211.0, 1074.0, 211.0, 1074.0, 196.0, 1048.0, 196.0], "text": "TM", "label": 25}, {"box": [535.0, 239.0, 833.0, 239.0, 833.0, 200.0, 535.0, 200.0], "text": "ATOREMGRTOMMILAZZO", "label": 25}, {"box": [907.0, 256.0, 1081.0, 256.0, 1081.0, 223.0, 907.0, 223.0], "text": "703-777-5833", "label": 5}, {"box": [576.0, 282.0, 1016.0, 282.0, 1016.0, 230.0, 576.0, 230.0], "text": "THANKYOUFORSHOPPINGWITHUS!", "label": 25}, {"box": [724.0, 387.0, 840.0, 387.0, 840.0, 355.0, 724.0, 355.0], "text": "GROCERY", "label": 25}, {"box": [474.0, 433.0, 733.0, 433.0, 733.0, 394.0, 474.0, 394.0], "text": "BCCHOCCUPCAKES", "label": 11}, {"box": [470.0, 466.0, 764.0, 466.0, 764.0, 421.0, 470.0, 421.0], "text": "ACNSPRYFRTSHAPE", "label": 11}, {"box": [466.0, 483.0, 595.0, 483.0, 595.0, 448.0, 466.0, 448.0], "text": "Regprice", "label": 25}, {"box": [463.0, 517.0, 658.0, 517.0, 658.0, 478.0, 463.0, 478.0], "text": "Cardsavings", "label": 25}, {"box": [459.0, 538.0, 574.0, 538.0, 574.0, 507.0, 459.0, 507.0], "text": "MfrCpn", "label": 25}, {"box": [455.0, 578.0, 686.0, 578.0, 686.0, 536.0, 455.0, 536.0], "text": "MultipliedCpn", "label": 25}, {"box": [453.0, 611.0, 750.0, 611.0, 750.0, 565.0, 453.0, 565.0], "text": "DULCEDELECHECHE", "label": 11}, {"box": [449.0, 625.0, 581.0, 625.0, 581.0, 593.0, 449.0, 593.0], "text": "Regprice", "label": 25}, {"box": [446.0, 662.0, 644.0, 662.0, 644.0, 623.0, 446.0, 623.0], "text": "CardSavings", "label": 25}, {"box": [443.0, 683.0, 558.0, 683.0, 558.0, 650.0, 443.0, 650.0], "text": "MfrCpn", "label": 25}, {"box": [441.0, 723.0, 673.0, 723.0, 673.0, 682.0, 441.0, 682.0], "text": "MuitipliedCpn", "label": 25}, {"box": [437.0, 758.0, 737.0, 758.0, 737.0, 708.0, 437.0, 708.0], "text": "MULTIGRAINCHEERIO", "label": 11}, {"box": [433.0, 774.0, 566.0, 774.0, 566.0, 739.0, 433.0, 739.0], "text": "RegPrice", "label": 25}, {"box": [430.0, 808.0, 632.0, 808.0, 632.0, 769.0, 430.0, 769.0], "text": "CardSavings", "label": 25}, {"box": [426.0, 830.0, 544.0, 830.0, 544.0, 800.0, 426.0, 800.0], "text": "MfrCpn", "label": 25}, {"box": [424.0, 872.0, 660.0, 872.0, 660.0, 831.0, 424.0, 831.0], "text": "MuitieliedCpn", "label": 25}, {"box": [419.0, 907.0, 708.0, 907.0, 708.0, 860.0, 419.0, 860.0], "text": "PROGRESSOR&HSTK", "label": 11}, {"box": [414.0, 922.0, 550.0, 922.0, 550.0, 890.0, 414.0, 890.0], "text": "RegPrice", "label": 25}, {"box": [413.0, 962.0, 615.0, 962.0, 615.0, 923.0, 413.0, 923.0], "text": "CardSavings", "label": 25}, {"box": [613.0, 1049.0, 836.0, 1049.0, 836.0, 1001.0, 613.0, 1001.0], "text": "REFRIG/FROZEN", "label": 25}, {"box": [1005.0, 465.0, 1070.0, 465.0, 1070.0, 437.0, 1005.0, 437.0], "text": "3.99", "label": 15}, {"box": [1023.0, 491.0, 1067.0, 491.0, 1067.0, 469.0, 1023.0, 469.0], "text": ".49", "label": 15}, {"box": [848.0, 509.0, 913.0, 509.0, 913.0, 481.0, 848.0, 481.0], "text": "3.79", "label": 25}, {"box": [847.0, 538.0, 927.0, 538.0, 927.0, 512.0, 847.0, 512.0], "text": "1.80-", "label": 25}, {"box": [863.0, 566.0, 925.0, 566.0, 925.0, 543.0, 863.0, 543.0], "text": ".75-", "label": 25}, {"box": [862.0, 596.0, 923.0, 596.0, 923.0, 572.0, 862.0, 572.0], "text": ".75-", "label": 25}, {"box": [993.0, 641.0, 1060.0, 641.0, 1060.0, 610.0, 993.0, 610.0], "text": "1.99", "label": 15}, {"box": [836.0, 655.0, 901.0, 655.0, 901.0, 629.0, 836.0, 629.0], "text": "4.19", "label": 25}, {"box": [835.0, 683.0, 916.0, 683.0, 916.0, 658.0, 835.0, 658.0], "text": "1.20-", "label": 25}, {"box": [853.0, 713.0, 914.0, 713.0, 914.0, 688.0, 853.0, 688.0], "text": ".50-", "label": 25}, {"box": [851.0, 743.0, 912.0, 743.0, 912.0, 719.0, 851.0, 719.0], "text": ".50-", "label": 25}, {"box": [985.0, 791.0, 1050.0, 791.0, 1050.0, 760.0, 985.0, 760.0], "text": "1.50", "label": 15}, {"box": [825.0, 804.0, 891.0, 804.0, 891.0, 775.0, 825.0, 775.0], "text": "3.99", "label": 25}, {"box": [826.0, 832.0, 906.0, 832.0, 906.0, 805.0, 826.0, 805.0], "text": "1.49-", "label": 25}, {"box": [840.0, 865.0, 903.0, 865.0, 903.0, 838.0, 840.0, 838.0], "text": ".50-", "label": 25}, {"box": [839.0, 895.0, 901.0, 895.0, 901.0, 869.0, 839.0, 869.0], "text": ".50-", "label": 25}, {"box": [972.0, 946.0, 1041.0, 946.0, 1041.0, 913.0, 972.0, 913.0], "text": "2.00", "label": 15}, {"box": [812.0, 957.0, 878.0, 957.0, 878.0, 928.0, 812.0, 928.0], "text": "2.69", "label": 25}, {"box": [831.0, 989.0, 894.0, 989.0, 894.0, 963.0, 831.0, 963.0], "text": ".69-", "label": 25}, {"box": [402.0, 1069.0, 483.0, 1069.0, 483.0, 1044.0, 402.0, 1044.0], "text": "1OIY", "label": 14}, {"box": [395.0, 1127.0, 703.0, 1127.0, 703.0, 1072.0, 395.0, 1072.0], "text": "PLSBRYBSCTSANDWI", "label": 11}, {"box": [395.0, 1159.0, 702.0, 1159.0, 702.0, 1105.0, 395.0, 1105.0], "text": "TOTINOSPZASTOFFE", "label": 11}, {"box": [390.0, 1173.0, 526.0, 1173.0, 526.0, 1136.0, 390.0, 1136.0], "text": "RegPrice", "label": 25}, {"box": [384.0, 1199.0, 589.0, 1199.0, 589.0, 1172.0, 384.0, 1172.0], "text": "", "label": 0}, {"box": [790.0, 1198.0, 861.0, 1198.0, 861.0, 1177.0, 790.0, 1177.0], "text": "", "label": 0}, {"box": [958.0, 1171.0, 1029.0, 1171.0, 1029.0, 1132.0, 958.0, 1132.0], "text": "3.50", "label": 15}, {"box": [956.0, 1198.0, 1029.0, 1198.0, 1029.0, 1163.0, 956.0, 1163.0], "text": "3.49", "label": 15}, {"box": [1091.0, 466.0, 1103.0, 466.0, 1103.0, 439.0, 1091.0, 439.0], "text": "B", "label": 25}, {"box": [1088.0, 494.0, 1102.0, 494.0, 1102.0, 469.0, 1088.0, 469.0], "text": "B", "label": 25}, {"box": [1078.0, 642.0, 1092.0, 642.0, 1092.0, 615.0, 1078.0, 615.0], "text": "B", "label": 25}, {"box": [1070.0, 792.0, 1087.0, 792.0, 1087.0, 765.0, 1070.0, 765.0], "text": "B", "label": 25}, {"box": [1062.0, 948.0, 1077.0, 948.0, 1077.0, 919.0, 1062.0, 919.0], "text": "B", "label": 25}, {"box": [1045.0, 1168.0, 1063.0, 1168.0, 1063.0, 1143.0, 1045.0, 1143.0], "text": "B", "label": 25}, {"box": [1044.0, 1198.0, 1060.0, 1198.0, 1060.0, 1176.0, 1044.0, 1176.0], "text": "B", "label": 25}]}
{"file_name": "tests/data/kie_toy_dataset/wildreceipt/2.jpeg", "height": 348, "width": 348, "annotations": [{"box": [114.0, 19.0, 230.0, 19.0, 230.0, 1.0, 114.0, 1.0], "text": "CHOEUN", "label": 1}, {"box": [97.0, 35.0, 236.0, 35.0, 236.0, 19.0, 97.0, 19.0], "text": "KOREANRESTAURANT", "label": 1}, {"box": [29.0, 56.0, 295.0, 56.0, 295.0, 34.0, 29.0, 34.0], "text": "2621ORANGETHORPEAVE,FULLERTON.", "label": 3}, {"box": [48.0, 73.0, 280.0, 73.0, 280.0, 54.0, 48.0, 54.0], "text": "(714)879-3574", "label": 5}, {"box": [79.0, 92.0, 259.0, 92.0, 259.0, 74.0, 79.0, 74.0], "text": "THANKYOU!!", "label": 25}, {"box": [22.0, 130.0, 61.0, 130.0, 61.0, 112.0, 22.0, 112.0], "text": "DATE", "label": 8}, {"box": [70.0, 131.0, 192.0, 131.0, 192.0, 112.0, 70.0, 112.0], "text": "12/30/2016FRI", "label": 7}, {"box": [263.0, 128.0, 307.0, 128.0, 307.0, 111.0, 263.0, 111.0], "text": "19:19", "label": 9}, {"box": [19.0, 168.0, 157.0, 168.0, 157.0, 149.0, 19.0, 149.0], "text": "BIBIM.OCTOPUT1", "label": 11}, {"box": [17.0, 190.0, 158.0, 190.0, 158.0, 171.0, 17.0, 171.0], "text": "S-FOODP.CAKT1", "label": 11}, {"box": [14.0, 214.0, 158.0, 214.0, 158.0, 192.0, 14.0, 192.0], "text": "PORKDUMPLINT1", "label": 11}, {"box": [14.0, 236.0, 151.0, 236.0, 151.0, 215.0, 14.0, 215.0], "text": "LABEEFRIBT1", "label": 11}, {"box": [254.0, 168.0, 312.0, 168.0, 312.0, 149.0, 254.0, 149.0], "text": "$13.99", "label": 15}, {"box": [257.0, 189.0, 314.0, 189.0, 314.0, 170.0, 257.0, 170.0], "text": "$14.99", "label": 15}, {"box": [268.0, 212.0, 316.0, 212.0, 316.0, 191.0, 268.0, 191.0], "text": "$8.99", "label": 15}, {"box": [261.0, 234.0, 318.0, 234.0, 318.0, 213.0, 261.0, 213.0], "text": "\uffe517.99", "label": 15}, {"box": [118.0, 260.0, 217.0, 260.0, 217.0, 239.0, 118.0, 239.0], "text": "4.00xITEMS", "label": 12}, {"box": [8.0, 285.0, 91.0, 285.0, 91.0, 264.0, 8.0, 264.0], "text": "SUBTOTAL", "label": 18}, {"box": [8.0, 312.0, 49.0, 312.0, 49.0, 291.0, 8.0, 291.0], "text": "TAX1", "label": 20}, {"box": [8.0, 336.0, 61.0, 336.0, 61.0, 316.0, 8.0, 316.0], "text": "TOTAL", "label": 24}, {"box": [263.0, 283.0, 325.0, 283.0, 325.0, 260.0, 263.0, 260.0], "text": "$55.96", "label": 17}, {"box": [274.0, 308.0, 326.0, 308.0, 326.0, 286.0, 274.0, 286.0], "text": "$4.48", "label": 19}, {"box": [267.0, 334.0, 328.0, 334.0, 328.0, 310.0, 267.0, 310.0], "text": "$60.44", "label": 23}, {"box": [269.0, 347.0, 328.0, 347.0, 328.0, 336.0, 269.0, 336.0], "text": "", "label": 0}, {"box": [11.0, 347.0, 50.0, 347.0, 50.0, 342.0, 11.0, 342.0], "text": "", "label": 0}, {"box": [215.0, 128.0, 253.0, 128.0, 253.0, 112.0, 215.0, 112.0], "text": "TIME", "label": 10}]}