File size: 832 Bytes
152f0f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
# Copyright (c) OpenMMLab. All rights reserved.
from unittest import TestCase
import numpy as np
from mmcv import imread
from mmpose.datasets.transforms.loading import LoadImage
class TestLoadImage(TestCase):
def test_load_image(self):
transform = LoadImage()
results = dict(img_path='tests/data/coco/000000000785.jpg')
results = transform(results)
self.assertIsInstance(results['img'], np.ndarray)
def test_with_input_image(self):
transform = LoadImage(to_float32=True)
img_path = 'tests/data/coco/000000000785.jpg'
results = dict(
img_path=img_path, img=imread(img_path).astype(np.uint8))
results = transform(results)
self.assertIsInstance(results['img'], np.ndarray)
self.assertTrue(results['img'].dtype, np.float32)
|