ValueError:检查输入时出错:预期密集型11_输入具有3个维度,但得到的数组具有形状(0,1)

2024-05-08 01:48:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我是机器学习和python的新手,我尝试对病人是否患有癌症进行分类。我从https://github.com/fahomid/ML-Tensorflow-Medical-Image/blob/master/tensorflow-model.py找到了一段代码。我有一个小数据集。Training和test set有两个病人目录,其中包含dicom文件,仅供尝试。代码如下:

import keras
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation, Flatten
from keras.layers import Conv2D, MaxPooling2D
import os
import pydicom
import numpy as np
import PIL

# Generate data from dicom file
dataset = [];
labels = [];
for root, dirs, files in os.walk("training_data/Cancer"):
    for file in files:
        if file.endswith(".dcm"):
            ds = pydicom.dcmread(os.path.join(root, file))
            dataset.append(ds.pixel_array)
            labels.append(1);

for root, dirs, files in os.walk("training_data/Normal"):
   for file in files:
        if file.endswith(".dcm"):
            ds = pydicom.dcmread(os.path.join(root, file))
            dataset.append(ds.pixel_array)
            labels.append(0)


dataset_size = len(dataset)
dataset = np.asarray(dataset)
labels = np.asarray(labels)

# create model
model = Sequential()
model.add(Dense(32, activation='tanh', input_shape=(512, 512)))
model.add(Flatten())
model.add(Dense(1, activation='sigmoid'))
model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics= 
['accuracy'])
model.fit(dataset, labels, epochs=10, shuffle=True, batch_size=32)

# serialize model to JSON
model_json = model.to_json()
with open("model.json", "w") as json_file:
    json_file.write(model_json)
# serialize weights to HDF5
model.save_weights("model.h5")
print("\n\nModel saved to disk\n\n")
model.summary()

错误信息行如下:

model.fit(dataset, labels, epochs=10, shuffle=True, batch_size=32)

ValueError: Error when checking input: expected dense_11_input to have 3 dimensions, but got array with shape (0, 1)

谢谢你的帮助。在


Tags: toinfromimportjsonforlabelsmodel
1条回答
网友
1楼 · 发布于 2024-05-08 01:48:53

代码的以下部分查找包含癌症正常人数据的文件(扩展名为.dcm)。它不查找任何内容,因此不返回任何内容。在

# Generate data from dicom file
dataset = [];
labels = [];
for root, dirs, files in os.walk("training_data/Cancer"):
    for file in files:
        if file.endswith(".dcm"):
            ds = pydicom.dcmread(os.path.join(root, file))
            dataset.append(ds.pixel_array)
            labels.append(1);

for root, dirs, files in os.walk("training_data/Normal"):
   for file in files:
        if file.endswith(".dcm"):
            ds = pydicom.dcmread(os.path.join(root, file))
            dataset.append(ds.pixel_array)
            labels.append(0)

所以dataset变量的值是0,而labels变量的值是1。当调用model.fit方法时,它期望输入是三维的,形状为(512, 512),但它只得到形状为(0, 1)的输入。在

相关问题 更多 >