|
import tempfile |
|
|
|
import cv2 |
|
import numpy as np |
|
import pytest |
|
from PIL import Image |
|
|
|
INPUT_HEIGHT, INPUT_WIDTH = 600, 500 |
|
|
|
|
|
@pytest.mark.parametrize('dtype,extension,dim', [ |
|
(np.uint8, '.png', 1), |
|
(np.uint8, '.jpg', 3), |
|
(np.uint8, '.png', 1), |
|
(np.uint16, '.png', 1), |
|
(np.uint16, '.jpg', 1), |
|
(np.float32, '.tiff', 1), |
|
(np.uint8, '.tiff', 3), |
|
] |
|
) |
|
def test_image_reader(dtype, extension, dim): |
|
tf = tempfile.NamedTemporaryFile(suffix=extension) |
|
|
|
if dtype in [np.uint8, np.uint16]: |
|
image_w = np.random.randint(0, 225, (225, 225, dim)).astype(dtype) |
|
else: |
|
image_w = np.random.rand(225, 225, dim).astype(dtype) |
|
|
|
if dim == 1: |
|
image_w = np.squeeze(image_w) |
|
cv2.imwrite(tf.name, image_w) |
|
|
|
image_r = np.array(Image.open(tf.name)) |
|
image_r = cv2.normalize( |
|
image_r, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U) |
|
image_r = cv2.resize( |
|
image_r, (INPUT_WIDTH, INPUT_HEIGHT), interpolation=cv2.INTER_LINEAR) |
|
|
|
assert image_r.dtype == np.uint8 |
|
assert image_r.shape[:2] == (INPUT_HEIGHT, INPUT_WIDTH) |
|
assert np.min(image_r) >= 0 and np.max(image_r) <= 255 |
|
|
