Bläddra i källkod

fix: 回退829

Zhang Li 1 år sedan
förälder
incheckning
0e62fb7a79
2 ändrade filer med 7 tillägg och 5 borttagningar
  1. 3 1
      server.py
  2. 4 4
      testing/table_test.py

+ 3 - 1
server.py

@@ -39,9 +39,10 @@ table_engine = PPStructure(layout=False,
                            table=True,
                            use_gpu=True,
                            show_log=True,
+                           use_angle_cls=True,
                            # det_model_dir="models/det/det_table_v2",
                            # rec_model_dir="models/rec/rec_table_v1",
-                           table_model_dir="models/table/SLANet_905")
+                           table_model_dir="models/table/SLANet_829")
 
 
 cls_lock = threading.Lock()
@@ -142,6 +143,7 @@ def table_res(im, ROTATE=-1):
     im = im.copy()
     # 获取正向图片
     img = get_zero_degree_image(im)
+    # cv2.imwrite('1.jpg', img)
     try:
         table_engine_lock.acquire()
         res = table_engine(img)

+ 4 - 4
testing/table_test.py

@@ -22,10 +22,10 @@ class TestTableOcr(unittest.TestCase):
     #     pong = requests.request("GET", '/'.join([url, 'ping']))
     #     self.assertEqual('"pong!"', pong.text, 'Not work')
 
-    def test_table_01(self):
-        print("270")
-        fn = Path(__file__).parent / '../images/01.jpeg'
-        res = send_request(fn)
+    # def test_table_01(self):
+    #     print("270")
+    #     fn = Path(__file__).parent / '../images/01.jpeg'
+    #     res = send_request(fn)
         # self.assertEqual('000', res['status'], 'Not work')
 
     def test_table_02(self):