我一直试图从CIFAR-10列车上删除特定的图像,但还没有成功。到目前为止,我试过了她的方法:
data = tfds.as_numpy(tfds.load(name=FLAGS.dataset, batch_size=-1, data_dir=DATA_DIR))
inputs = data['train']['image']
labels = data['train']['label']
inputs = (inputs/127.5)-1
inputs = np.delete(inputs, [0, 4, 3, 2])
labels = np.delete(labels, [0, 4, 3, 2])
我有一个特定图像[0, 4, 3, 2]
的索引列表,我想要删除。inputs
的形状为(50000, 32, 32, 3)
,labels
的形状为(50000,)
。上面的方法根本不起作用,因为出于某种原因,input
的形状变得近5倍大。如果能帮上忙,我将不胜感激。
1条答案
按热度按时间63lcw9qa1#
Np.ete says的文档:
轴:整型,可选
要沿其删除由obj定义的子数组的轴。如果Axis为None,则将obj应用于展平的数组。
因此,在删除元素时指定样本轴