| 47 | return imsave(inverse_transform(images), size, image_path) |
| 48 | |
| 49 | def imread(path, grayscale = False): |
| 50 | if (grayscale): |
| 51 | return scipy.misc.imread(path, flatten = True).astype(np.float) |
| 52 | else: |
| 53 | # Reference: https://github.com/carpedm20/DCGAN-tensorflow/issues/162#issuecomment-315519747 |
| 54 | img_bgr = cv2.imread(path) |
| 55 | # Reference: https://stackoverflow.com/a/15074748/ |
| 56 | img_rgb = img_bgr[..., ::-1] |
| 57 | return img_rgb.astype(np.float) |
| 58 | |
| 59 | def merge_images(images, size): |
| 60 | return inverse_transform(images) |