批次大小大于1的产品不适用于Cyclegan

时间:2018-07-11 16:41:51

标签: python tensorflow

我正在使用tensorflow用于python的python代码,如https://github.com/gitlimlab/CycleGAN-Tensorflow所示。我将两个输入堆叠如下:

image_a = np.stack(data_A[iter*self._batch_size:(iter+1)*self._batch_size])
image_b = np.stack(data_B[iter*self._batch_size:(iter+1)*self._batch_size])

当我使用大于1的批处理大小时,出现如下错误

image_a = np.stack(data_A [iter * self._batch_size:(iter + 1)* self._batch_size])

文件“ /NumPy/1.14.5/osl42-x86_64/lib/python2.7/site-packages/numpy/core/shape_base.py”,第349行,在堆栈中 引发ValueError('至少需要堆叠一个数组')

ValueError:至少需要堆叠一个数组

有人可以指导我吗? 预先感谢

0 个答案:

没有答案