| # -*- coding: utf-8 -*- | |
| import numpy as np | |
| def _get_pixels(per_pixel, rand_color, patch_size, dtype=np.float32): | |
| if per_pixel: | |
| return np.random.randint(0, 255, patch_size).astype(dtype=dtype) | |
| elif rand_color: | |
| return np.random.randint(0, 255, (1, 1, patch_size[2])).astype(dtype=dtype) | |
| else: | |
| return np.zeros((1, 1, patch_size[2]), dtype=dtype) | |