CT图片分类–AlexNet

CT图片分类–AlexNet

链接：https://pan.baidu.com/s/15CxZVg_3L466hkpgeQ7-VA
提取码：dwck
ct图片的链接

from PIL import Image#图像进行操作
import matplotlib.pyplot as plt#绘图
import numpy as np#处理数组、效率很高
import os#处理文件、读文件
import tensorflow as tf
from tensorflow.keras import datasets, layers, models#kears是一个高级api

加载图片

加载livergate

X_train = Image.open('C:/Users/31035/Desktop/Desktop/data/livergate/livergate_0.png')
X_train = X_train.resize((512,512),Image.ANTIALIAS)
X_train = np.array(X_train)
X_train = np.expand_dims(X_train, axis=0)
print(X_train.shape)
for dirname, _, filenames in os.walk('C:/Users/31035/Desktop/Desktop/data/livergate'):
    for filename in filenames:
        if X_train.shape[0] > 357:
            break
        try:
            im = Image.open(os.path.join(dirname, filename))
            im = im.resize((512,512),Image.ANTIALIAS)
            image_array = np.array(im)
            image_array = np.expand_dims(image_array, axis=0)
            X_train = np.concatenate((X_train, image_array), axis=0)
        except:
            pass
print(str(X_train.shape[0]))

加载nolivergate、

X_train_n = Image.open('C:/Users/31035/Desktop/Desktop/data/nolivergate/other_0.png')
X_train_n = X_train_n.resize((512,512),Image.ANTIALIAS)
X_train_n = np.array(X_train_n)
X_train_n = np.expand_dims(X_train_n, axis=0)
print(X_train_n.shape)

for dirname, _, filenames in os.walk('C:/Users/31035/Desktop/Desktop/data/nolivergate'):
    for filename in filenames:
        if X_train_n.shape[0] > 820:
            break
        try:
            im = Image.open(os.path.join(dirname, filename))
            im = im.resize((512,512),Image.ANTIALIAS)
            image_array = np.array(im)
            image_array = np.expand_dims(image_array, axis=0)
            X_train_n = np.concatenate((X_train_n, image_array), axis=0)
        except:
            pass
print(X_train_n.shape)

设置标签

Y_train = np.ones((358))
Y_train = Y_train.astype(int)
Y_train_n = np.zeros((821))
Y_train_n = Y_train_n.astype(int)

合并数据后打乱

#如果不把数据集打乱，特征、标签集中，验证集效率会很低
X = np.concatenate((X_train,X_train_n),axis=0)
Y = np.concatenate((Y_train,Y_train_n),axis=0)
np.random.seed(200)
np.random.shuffle(X)
np.random.seed(200)
np.random.shuffle(Y)
X.shape,X.dtype

展示图片

class_names = ['nolivergate','livergate']

plt.figure(figsize=(10,10))
for i in range(25):
    plt.subplot(5,5,i+1)
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow(X[i]，cmap='gray')
        plt.xlabel(class_names[Y[i]])
plt.show()

数据集后续处理

#给数据集增加一个维度，方便后面神经网络处理
X = tf.expand_dims(input=X,axis=-1)
#归一化，在这里归一化可以省略
X/255
#分离训练测试集
X_tra = X[:800]
X_test = X[800:]
y_tra = Y[:800]
Y_test = Y[800:]

建模分析预测

def AlexNet_inference(in_shape):
    model = keras.Sequential(name='AlexNet')  
 
    # model.add(layers.Conv2D(96,(11,11),strides=(4,4),input_shape=(in_shape[1],in_shape[2],in_shape[3]),
                # padding='same',activation='relu',kernel_initializer='uniform')) 
                
    model.add(layers.Conv2D(96,(11,11),strides=(2,2),input_shape=(in_shape[1],in_shape[2],in_shape[3]),
                padding='same',activation='relu',kernel_initializer='uniform'))  
    model.add(layers.MaxPooling2D(pool_size=(3,3),strides=(2,2)))  
    model.add(layers.Conv2D(256,(5,5),strides=(1,1),padding='same',activation='relu',kernel_initializer='uniform'))  
    model.add(layers.MaxPooling2D(pool_size=(3,3),strides=(2,2)))  
    model.add(layers.Conv2D(384,(3,3),strides=(1,1),padding='same',activation='relu',kernel_initializer='uniform'))  
    model.add(layers.Conv2D(384,(3,3),strides=(1,1),padding='same',activation='relu',kernel_initializer='uniform'))  
    model.add(layers.Conv2D(256,(3,3),strides=(1,1),padding='same',activation='relu',kernel_initializer='uniform'))  
    model.add(layers.MaxPooling2D(pool_size=(2,2),strides=(2,2)))  
    
    model.add(layers.Flatten())  
    model.add(layers.Dense(64,activation='relu'))  
    model.add(layers.Dropout(0.5))  
    model.add(layers.Dense(64,activation='relu'))  
    model.add(layers.Dropout(0.5))  
    model.add(layers.Dense(2,activation='softmax'))  
    model.compile(optimizer=keras.optimizers.Adam(),
                loss='sparse_categorical_crossentropy', #不能直接用函数，否则在与测试加载模型不成功！
                metrics=['accuracy'])
    model.summary()  
    return model

x_shape  = X_tra.shape
AlexNet_model = AlexNet_inference(x_shape)
totall_epochs = 0
epochs = 10

history = AlexNet_model.fit(X_tra, Y_tra, batch_size=10, epochs=epochs, validation_split=0.3)

plt.plot(history.history['accuracy'])
plt.plot(history.history['val_accuracy'])
plt.legend(['training', 'valivation'], loc='upper left')
plt.show()
res = AlexNet_model.evaluate(X_test, Y_test)
print(res)

测试集结果

继续训练10轮

history = AlexNet_model.fit(X_tra, Y_tra, batch_size=10, epochs=epochs, validation_split=0.3)

plt.plot(history.history['accuracy'])
plt.plot(history.history['val_accuracy'])
plt.legend(['training', 'valivation'], loc='upper left')
plt.show()
res = AlexNet_model.evaluate(X_test, Y_test)
print(res)