Skip to content

Commit

Permalink
chore(rapidocr_openvino/paddle): Optim code
Browse files Browse the repository at this point in the history
  • Loading branch information
SWHL committed Sep 19, 2024
1 parent 64da0dd commit 81cea70
Show file tree
Hide file tree
Showing 30 changed files with 12 additions and 12 deletions.
6 changes: 3 additions & 3 deletions python/rapidocr_openvino/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import cv2
import numpy as np

from .ch_ppocr_v2_cls import TextClassifier
from .ch_ppocr_v3_det import TextDetector
from .ch_ppocr_v3_rec import TextRecognizer
from .ch_ppocr_cls import TextClassifier
from .ch_ppocr_det import TextDetector
from .ch_ppocr_rec import TextRecognizer
from .utils import (
LoadImage,
UpdateParameters,
Expand Down
6 changes: 3 additions & 3 deletions python/rapidocr_paddle/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
import cv2
import numpy as np

from .ch_ppocr_v2_cls import TextClassifier
from .ch_ppocr_v3_det import TextDetector
from .ch_ppocr_v3_rec import TextRecognizer
from .ch_ppocr_cls import TextClassifier
from .ch_ppocr_det import TextDetector
from .ch_ppocr_rec import TextRecognizer
from .utils import (
LoadImage,
UpdateParameters,
Expand Down
6 changes: 3 additions & 3 deletions python/tests/test_paddle.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def test_input_three_ndim_one_channel():


def test_det():
module_name = "ch_ppocr_v3_det"
module_name = "ch_ppocr_det"
class_name = "TextDetector"

base = BaseModule(package_name)
Expand All @@ -212,7 +212,7 @@ def test_det():


def test_cls():
module_name = "ch_ppocr_v2_cls"
module_name = "ch_ppocr_cls"
class_name = "TextClassifier"

base = BaseModule(package_name=package_name)
Expand All @@ -231,7 +231,7 @@ def test_cls():


def test_rec():
module_name = "ch_ppocr_v3_rec"
module_name = "ch_ppocr_rec"
class_name = "TextRecognizer"

base = BaseModule(package_name)
Expand Down
6 changes: 3 additions & 3 deletions python/tests/test_vino.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def test_input_three_ndim_one_channel():


def test_det():
module_name = "ch_ppocr_v3_det"
module_name = "ch_ppocr_det"
class_name = "TextDetector"

base = BaseModule(package_name)
Expand All @@ -212,7 +212,7 @@ def test_det():


def test_cls():
module_name = "ch_ppocr_v2_cls"
module_name = "ch_ppocr_cls"
class_name = "TextClassifier"

base = BaseModule(package_name=package_name)
Expand All @@ -231,7 +231,7 @@ def test_cls():


def test_rec():
module_name = "ch_ppocr_v3_rec"
module_name = "ch_ppocr_rec"
class_name = "TextRecognizer"

base = BaseModule(package_name)
Expand Down

0 comments on commit 81cea70

Please sign in to comment.