水珠（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))將在self.dataset_name子目錄./data返回任何.npy文件名。