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 数据（它返回空白，即[]）。

因此，我想知道对于 numpy 文件，glob(os.path.join()) 有什么好的替代品？或者是否有任何参数可以使 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)) 将返回./data 的self.dataset_name 子目录中的任何.npy 文件名。