我一直在嘗試從 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]我想洗掉的特定影像的索引串列。is的形狀和inputsis(50000, 32, 32, 3)的形狀。上述方法根本不起作用,因為由于某種原因,形狀變大了近 5 倍。我很感激這里的任何幫助。labels(50000,)input
uj5u.com熱心網友回復:
np.delete 的檔案說:
軸:int,可選
沿其洗掉由 obj 定義的子陣列的軸。如果axis為None,則將obj應用于展平陣列。
所以在洗掉元素時指定樣本的軸
inputs = np.delete(inputs, [0, 4, 3, 2], axis=0)
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/529640.html
上一篇:如何在回圈中命令Ajax呼叫
