mirror of https://github.com/open-mmlab/mmpose
32 lines
832 B
Python
32 lines
832 B
Python
# 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)
|