glob(os.path.join()) для работы с данными.npy

Question

Я пытаюсь дополнить код DC-GANS, чтобы он работал с моими данными. Исходный код имеет данные в формате JPEG, однако я бы очень хотел, чтобы мои данные были в формате.npy.

Проблема в строке 76: self.data = glob(os.path.join("./data", self.dataset_name, self.input_fname_pattern)) не будет работать с пустыми данными (возвращается пустым, т.е. []).

Поэтому мне интересно, что является хорошей заменой glob(os.path.join()) для обалденных файлов? Или есть какие-нибудь параметры, которые сделали бы glob совместимым с бесчисленными данными?

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 имена файлов в self.dataset_name подкаталог ./data,