2017-05-28 75 views
0

我试着去增加DC-GANS code使其作品与我的数据。原始代码的数据为JPEG,但我真的强烈希望将我的数据放在.npy中。水珠(os.path.join())与.npy数据

问题是行76:self.data = glob(os.path.join("./data", self.dataset_name, self.input_fname_pattern))将无法​​使用numpy数据(它回来空白,即[])。

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

回答

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))将在self.dataset_name子目录./data返回任何.npy文件名。