mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
44ca9c2a61
* Remove usage of \ Signed-off-by: lizz <lizz@sensetime.com> * rebase Signed-off-by: lizz <lizz@sensetime.com> * typos Signed-off-by: lizz <lizz@sensetime.com> * Remove test dependency on tools/ Signed-off-by: lizz <lizz@sensetime.com> * Remove usage of \ Signed-off-by: lizz <lizz@sensetime.com> * rebase Signed-off-by: lizz <lizz@sensetime.com> * typos Signed-off-by: lizz <lizz@sensetime.com> * Remove test dependency on tools/ Signed-off-by: lizz <lizz@sensetime.com> * typo Signed-off-by: lizz <lizz@sensetime.com> * KIE in keywords Signed-off-by: lizz <lizz@sensetime.com> * some renames Signed-off-by: lizz <lizz@sensetime.com> * kill isort skip Signed-off-by: lizz <lizz@sensetime.com> * aggregation discrimination Signed-off-by: lizz <lizz@sensetime.com> * aggregation discrimination Signed-off-by: lizz <lizz@sensetime.com> * tiny Signed-off-by: lizz <lizz@sensetime.com> * fix bug: model infer on cpu Co-authored-by: Hongbin Sun <hongbin306@gmail.com>
19 lines
494 B
Python
19 lines
494 B
Python
"""Test orientation check and ignore method."""
|
|
|
|
import shutil
|
|
import tempfile
|
|
|
|
from mmocr.utils import drop_orientation
|
|
|
|
|
|
def test_drop_orientation():
|
|
img_file = 'tests/data/test_img2.jpg'
|
|
output_file = drop_orientation(img_file)
|
|
assert output_file is img_file
|
|
|
|
img_file = 'tests/data/test_img1.jpg'
|
|
tmp_dir = tempfile.TemporaryDirectory()
|
|
dst_file = shutil.copy(img_file, tmp_dir.name)
|
|
output_file = drop_orientation(dst_file)
|
|
assert output_file[-3:] == 'png'
|