ValueError: невозможно преобразовать массив размера 10 в форму (10,3,32,32) - PullRequest
0 голосов
/ 20 февраля 2019

Я получаю ошибку изменения формы, и я не уверен, как ее исправить.Вот что я сделал до сих пор.Любой совет, что я сделал не так?Спасибо!

import pickle
import numpy as np
import matplotlib.pyplot as plt
import os

dataset_folder_path = 'cifar-10-batches-py'

labels = []
data = []
alllabels = []
alldata = []

    for i in range(1,6): 
        print("procesing file ", '/data_batch_' + str(i))
        with open(dataset_folder_path + '/data_batch_' + str(i), mode='rb')as file:
        my_data = (pickle.load(file, encoding='latin1'))        
        labels = my_data['labels']
        alllabels = np.append(alllabels,labels)

        data = my_data['data']
        alldata = np.append(alldata, data)


   def getbatch(alldata, alllabels, batch_size = 16):
        print(batch_size)
        for start in range(0, len(alldata), batch_size):
            end = min(start + batch_size, len(alldata))
            yield (alldata[start:end], alllabels[start:end])


    batchsize = 10
    for x, y in getbatch(alldata, alllabels, batchsize):
        x = x.reshape(batchsize, 3, 32, 32)    
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...