caffe_GAN
caffe_GAN copied to clipboard
你好,请问label的数量为什么初始化成batch_paths的两倍?
label = np.zeros((len(batch_paths) * 2,1), dtype=np.int)
label[:(len(batch_paths) - 1)] = 1
blobs['label'] = label``
输入是彩色图,dataLayer层会把彩色图片灰度化, 将彩色图片和灰度图一起输入,所以label是2倍,分别是彩色图的和灰度图的label
看明白了。另外提醒你一下,我看了你的代码,你调用cv2.resize的时候好像把宽度和高度弄反了,按照opencv的文档应该是宽度在前面,然后才是高度,虽然在这里图片高度和宽度相等,对结果没有影响,但是还是比较会产生一些误导的
谢谢!