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))将在.npy的{​​{1}}子目录中返回任何self.dataset_name个文件名。