mirror of https://github.com/open-mmlab/mmocr.git
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
||
import unittest.mock as mock
|
||
|
||
import numpy as np
|
||
|
||
import mmocr.datasets.pipelines.transforms as transforms
|
||
|
||
|
||
@mock.patch('%s.transforms.np.random.random_sample' % __name__)
|
||
def test_scale_aspect_jitter(mock_random):
|
||
img_scale = [(3000, 1000)] # unused
|
||
ratio_range = (0.5, 1.5)
|
||
aspect_ratio_range = (1, 1)
|
||
multiscale_mode = 'value'
|
||
long_size_bound = 2000
|
||
short_size_bound = 640
|
||
resize_type = 'long_short_bound'
|
||
keep_ratio = False
|
||
jitter = transforms.ScaleAspectJitter(
|
||
img_scale=img_scale,
|
||
ratio_range=ratio_range,
|
||
aspect_ratio_range=aspect_ratio_range,
|
||
multiscale_mode=multiscale_mode,
|
||
long_size_bound=long_size_bound,
|
||
short_size_bound=short_size_bound,
|
||
resize_type=resize_type,
|
||
keep_ratio=keep_ratio)
|
||
mock_random.side_effect = [0.5]
|
||
|
||
# test sample_from_range
|
||
|
||
result = jitter.sample_from_range([100, 200])
|
||
assert result == 150
|
||
|
||
# test _random_scale
|
||
results = {}
|
||
results['img'] = np.zeros((4000, 1000))
|
||
mock_random.side_effect = [0.5, 1]
|
||
jitter._random_scale(results)
|
||
# scale1 0.5， scale2=1 scale =0.5 650/1000, w, h
|
||
# print(results['scale'])
|
||
assert results['scale'] == (650, 2600)
|