StackGAN 泡菜错误

Question

我正在尝试从this repo 运行代码，这意味着我必须使这个get_data 函数工作：

def get_data(self, pickle_path, aug_flag=True):
    with open(pickle_path + self.image_filename, 'rb') as f:
        images = pickle.load(f)   # <--------- THIS line is the problem
        images = np.array(images)
        print('images: ', images.shape)
    # do more things here

但它给了我错误ValueError: unsupported pickle protocol: 3 ，所以我找到了建议here，他们推荐了不同的协议：pickle.dump(images, f, protocol=2)

def get_data(self, pickle_path, aug_flag=True):
    with open(pickle_path + self.image_filename, 'rb') as f:
        pickle.dump(images, f, protocol=2)   # still bad
        images = np.array(images)
        print('images: ', images.shape)
    # do more things here

但是，这给了我错误UnboundLocalError: local variable 'images' referenced before assignment。有没有办法解决这个问题，特别是StackGAN/misc/datasets.py？

Answer 1

我找到了答案here：

def get_data(self, pickle_path, aug_flag=True):
    with open(pickle_path + self.image_filename, 'rb') as f:
        pickle.dump(pickle.load(sys.stdin), sys.stdout, 2)  # <----- *****
        images = np.array(images)
        print('images: ', images.shape)
    # do more things here

StackGAN 提供用 Python3 腌制的文件，我需要将它们转换为 Python2 协议