glob(os.path.join()) 处理 .npy 数据

Question

我正在尝试扩充 DC-GANS code 以便它适用于我的数据。原始代码的数据为 JPEG，但是我真的非常希望将我的数据保存在 .npy 中。

问题是第 76 行：self.data = glob(os.path.join("./data", self.dataset_name, self.input_fname_pattern)) 无法处理 numpy 数据（它返回空白，即 []）。

因此我想知道什么是 glob(os.path.join()) 的 numpy 文件的良好替代品？或者是否有任何参数可以使 glob 与 numpy 数据兼容？

Answer 1

在DCGAN.__init__中，将input_fname_pattern='*.jpg'更改为input_fname_pattern='*.npy'：

class DCGAN(object):
  def __init__(self, ...
               input_fname_pattern='*.npy'...):

这会将 input_fname_pattern 的 默认值 更改为 '*.npy'。

或者，当您实例化 DCGAN 时，您可以将 input_fname_pattern='*.npy' 传递给它：

dcgan = DCGAN(sess, input_fname_pattern='*.npy')

如果你做这两件事中的一件，那么 glob(os.path.join("./data", self.dataset_name, self.input_fname_pattern)) 将 return ./data 的 self.dataset_name 子目录中的任何 .npy 文件名。

glob(os.path.join()) 处理 .npy 数据

glob(os.path.join()) to work with the .npy data

python

glob

numpy