值“ images.shape”不可取消

在dataset_tool.py中使用Nvlab StyleGan时,显示错误为:

value 'images.shape' is unsubscriptable

这是代码显示错误:

def create_cifar10(tfrecord_dir,cifar10_dir):
print('Loading CIFAR-10 from "%s"' % cifar10_dir)
import pickle
images = []
labels = []
for batch in range(1,6):
    with open(os.path.join(cifar10_dir,'data_batch_%d' % batch),'rb') as file:
        data = pickle.load(file,encoding='latin1')
    images.append(data['data'].reshape(-1,3,32,32))
    labels.append(data['labels'])
images = np.concatenate(images)
labels = np.concatenate(labels)
assert images.shape == (50000,32) and images.dtype == np.uint8
assert labels.shape == (50000,) and labels.dtype == np.int32
assert np.min(images) == 0 and np.max(images) == 255
assert np.min(labels) == 0 and np.max(labels) == 9
onehot = np.zeros((labels.size,np.max(labels) + 1),dtype=np.float32)
onehot[np.arange(labels.size),labels] = 1.0

with TFRecordExporter(tfrecord_dir,images.shape[0]) as tfr:
    order = tfr.choose_shuffled_order()
    for idx in range(order.size):
        tfr.add_image(images[order[idx]])
    tfr.add_labels(onehot[order])
henryjiangyun 回答:值“ images.shape”不可取消

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3128837.html

大家都在问