|
|
|
|
|
import numpy as np |
|
|
|
|
|
from mmocr.core import det_recog_show_result |
|
|
|
|
|
|
|
|
def test_det_recog_show_result(): |
|
|
img = np.ones((100, 100, 3), dtype=np.uint8) * 255 |
|
|
det_recog_res = { |
|
|
'result': [{ |
|
|
'box': [51, 88, 51, 62, 85, 62, 85, 88], |
|
|
'box_score': 0.9417, |
|
|
'text': 'hell', |
|
|
'text_score': 0.8834 |
|
|
}] |
|
|
} |
|
|
|
|
|
vis_img = det_recog_show_result(img, det_recog_res) |
|
|
|
|
|
assert vis_img.shape[0] == 100 |
|
|
assert vis_img.shape[1] == 200 |
|
|
assert vis_img.shape[2] == 3 |
|
|
|
|
|
det_recog_res['result'][0]['text'] = '中文' |
|
|
det_recog_show_result(img, det_recog_res) |
|
|
|
