numpy 如何从CIFAR-10中删除特定图像?

qgelzfjb  于 2022-11-10  发布在  其他
关注(0)|答案(1)|浏览(135)

我一直试图从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倍大。如果能帮上忙,我将不胜感激。

63lcw9qa

63lcw9qa1#

Np.ete says的文档:
轴:整型,可选
要沿其删除由obj定义的子数组的轴。如果Axis为None,则将obj应用于展平的数组。
因此,在删除元素时指定样本轴

inputs = np.delete(inputs, [0, 4, 3, 2], axis=0)

相关问题