AstarLight / Satellite-Segmentation

615 stars 233 forks source link

请问如何解决ValueError: cannot reshape array of size 9907170 into shape (65536,) #61

Open PingYufeng opened 4 years ago

PingYufeng commented 4 years ago

源码: def generateData(batch_size,data=[]):

print 'generateData...'

while True:  
    train_data = []  
    train_label = []  
    batch = 0  
    for i in (range(len(data))): 
        url = data[i]
        batch += 1 
        img = load_img(filepath + 'src/' + url)
        img = img_to_array(img) 
        train_data.append(img)  
        label = load_img(filepath + 'label/' + url, grayscale=True)
        **label = img_to_array(label).reshape((img_w*img_h,))**  
        print(label.shape)  
        train_label.append(label)  

报错:
  File "E:\RiseFile\Deeplearning\Satellite-Segmentation-master\segnet\segnet_train.py", line 78, in generateData
    label = img_to_array(label).reshape((img_w*img_h,))
ValueError: cannot reshape array of size 9907170 into shape (65536,)

img_to_array后无法reshape,是不是写错了?在windows下运行的,请问如何解决?
JoernZheng commented 4 years ago

可以复现的,你这个估计是导错图片了,这个错误是大小不匹配。 你在file填写的地址应该是256*256这个切割后大小的图片,而不是原始图片。