我正在使用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:至少需要堆叠一个数组
有人可以指导我吗? 预先感谢