質問編集履歴

補足しました。

2019/10/19 03:18

投稿

RyotaYamada

スコア6

test CHANGED Viewed

File without changes

test CHANGED Viewed

@@ -8,6 +8,38 @@
 ```
+ [*] Reading checkpoint...
+ [!] Load failed...
+Processing image: ./datasets/voice/testA\at.jpg
+Traceback (most recent call last):
+  File "main.py", line 53, in <module>
+    tf.app.run()
+  File "C:\Users\mounf\Anaconda3\envs\cycleganvoicechangerproject\lib\site-packages\tensorflow\python\platform\app.py", line 126, in run
+    _sys.exit(main(argv))
+  File "main.py", line 50, in main
+    else model.test(args)
+  File "C:\Users\mounf\PycharmProjects\nets\model.py", line 252, in test
+    sample_image = [load_test_data(sample_file, args.fine_size)]
+  File "C:\Users\mounf\PycharmProjects\nets\utils.py", line 46, in load_test_data
+    img = imread(image_path)
+  File "C:\Users\mounf\PycharmProjects\nets\utils.py", line 88, in imread
+    return _imread(path, mode='RGB').astype(np.float)
   File "C:\Users\mounf\Anaconda3\envs\cycleganvoicechangerproject\lib\site-packages\imageio\core\functions.py", line 260, in imread
     'Invalid keyword argument "mode", ' 'perhaps you mean "pilmode"?'
@@ -18,6 +50,8 @@
 ```
@@ -28,6 +62,306 @@
 ```python
+♯utils.py
+from __future__ import division
+import math
+import pprint
+import scipy.misc
+import numpy as np
+import copy
+try:
+    _imread = scipy.misc.imread
+except AttributeError:
+    from imageio import imread as _imread
+pp = pprint.PrettyPrinter()
+get_stddev = lambda x, k_h, k_w: 1/math.sqrt(k_w*k_h*x.get_shape()[-1])
+♯new added functions for cyclegan
+class ImagePool(object):
+    def __init__(self, maxsize=50):
+        self.maxsize = maxsize
+        self.num_img = 0
+        self.images = []
+    def __call__(self, image):
+        if self.maxsize <= 0:
+            return image
+        if self.num_img < self.maxsize:
+            self.images.append(image)
+            self.num_img += 1
+            return image
+        if np.random.rand() > 0.5:
+            idx = int(np.random.rand()*self.maxsize)
+            tmp1 = copy.copy(self.images[idx])[0]
+            self.images[idx][0] = image[0]
+            idx = int(np.random.rand()*self.maxsize)
+            tmp2 = copy.copy(self.images[idx])[1]
+            self.images[idx][1] = image[1]
+            return [tmp1, tmp2]
+        else:
+            return image
+def load_test_data(image_path, fine_size=256):
+    img = imread(image_path)
+    img = scipy.misc.imresize(img, [fine_size, fine_size])
+    img = img/127.5 - 1
+    return img
+def load_train_data(image_path, load_size=286, fine_size=256, is_testing=False):
+    img_A = imread(image_path[0])
+    img_B = imread(image_path[1])
+    if not is_testing:
+        img_A = scipy.misc.imresize(img_A, [load_size, load_size])
+        img_B = scipy.misc.imresize(img_B, [load_size, load_size])
+        h1 = int(np.ceil(np.random.uniform(1e-2, load_size-fine_size)))
+        w1 = int(np.ceil(np.random.uniform(1e-2, load_size-fine_size)))
+        img_A = img_A[h1:h1+fine_size, w1:w1+fine_size]
+        img_B = img_B[h1:h1+fine_size, w1:w1+fine_size]
+        if np.random.random() > 0.5:
+            img_A = np.fliplr(img_A)
+            img_B = np.fliplr(img_B)
+    else:
+        img_A = scipy.misc.imresize(img_A, [fine_size, fine_size])
+        img_B = scipy.misc.imresize(img_B, [fine_size, fine_size])
+    img_A = img_A/127.5 - 1.
+    img_B = img_B/127.5 - 1.
+    img_AB = np.concatenate((img_A, img_B), axis=2)
+    # img_AB shape: (fine_size, fine_size, input_c_dim + output_c_dim)
+    return img_AB
+-----------------------------
+def get_image(image_path, image_size, is_crop=True, resize_w=64, is_grayscale = False):
+    return transform(imread(image_path, is_grayscale), image_size, is_crop, resize_w)
+def save_images(images, size, image_path):
+    return imsave(inverse_transform(images), size, image_path)
+def imread(path, is_grayscale = False):
+    if (is_grayscale):
+        return _imread(path, flatten=True).astype(np.float)
+    else:
+        return _imread(path, mode='RGB').astype(np.float)
+def merge_images(images, size):
+    return inverse_transform(images)
+def merge(images, size):
+    h, w = images.shape[1], images.shape[2]
+    img = np.zeros((h * size[0], w * size[1], 3))
+    for idx, image in enumerate(images):
+        i = idx % size[1]
+        j = idx // size[1]
+        img[j*h:j*h+h, i*w:i*w+w, :] = image
+    return img
+def imsave(images, size, path):
+    return scipy.misc.imsave(path, merge(images, size))
+def center_crop(x, crop_h, crop_w,
+                resize_h=64, resize_w=64):
+  if crop_w is None:
+    crop_w = crop_h
+  h, w = x.shape[:2]
+  j = int(round((h - crop_h)/2.))
+  i = int(round((w - crop_w)/2.))
+  return scipy.misc.imresize(
+      x[j:j+crop_h, i:i+crop_w], [resize_h, resize_w])
+def transform(image, npx=64, is_crop=True, resize_w=64):
+    # npx : # of pixels width/height of image
+    if is_crop:
+        cropped_image = center_crop(image, npx, resize_w=resize_w)
+    else:
+        cropped_image = image
+    return np.array(cropped_image)/127.5 - 1.
+def inverse_transform(images):
+    return (images+1.)/2.
+♯---ここからfunction.py
+♯Images
+def imread(uri, format=None, **kwargs):
+    """ imread(uri, format=None, **kwargs)
+    Reads an image from the specified file. Returns a numpy array, which
+    comes with a dict of meta data at its 'meta' attribute.
+    Note that the image data is returned as-is, and may not always have
+    a dtype of uint8 (and thus may differ from what e.g. PIL returns).
+    Parameters
+    ♯----------
+    uri : {str, pathlib.Path, bytes, file}
+        The resource to load the image from, e.g. a filename, pathlib.Path,
+        http address or file object, see the docs for more info.
+    format : str
+        The format to use to read the file. By default imageio selects
+        the appropriate for you based on the filename and its contents.
+    kwargs : ...
+        Further keyword arguments are passed to the reader. See :func:`.help`
+        to see what arguments are available for a particular format.
+    """
     if "mode" in kwargs:
         raise TypeError(
@@ -36,7 +370,15 @@
         )
+    ♯Get reader and read first
+    reader = read(uri, format, "i", **kwargs)
-```
+    with reader:
+        return reader.get_data(0)