请参考下面的示例代码来实现Conv3D
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Dropout, Activation, Flatten
from tensorflow.keras.layers import Conv3D, MaxPooling3D, BatchNormalization
import numpy as np
input_shape=(224, 224, 3, 5)
model = Sequential()
#C1
model.add(Conv3D(16, (3, 3, 1), strides=(1, 2, 2), padding='same',activation='relu',data_format= "channels_first", input_shape=input_shape))
model.add(MaxPooling3D(pool_size=(1, 2, 2),data_format= "channels_first", padding='same'))
model.add(BatchNormalization())
#C2
model.add(Conv3D(32, (3, 3, 1), strides=(1, 1, 1), padding='same',data_format= "channels_first", activation='relu'))
model.add(MaxPooling3D(pool_size=(1, 2, 2),data_format= "channels_first", padding='same'))
model.add(BatchNormalization())
#C3
model.add(Conv3D(64, (3, 3, 1), strides=(1, 1, 1), padding='same',data_format= "channels_first", activation='relu'))
model.add(MaxPooling3D(pool_size=(1, 2, 2), data_format= "channels_first", padding='same'))
model.add(BatchNormalization())
model.add(Flatten())
model.add(Dropout(0.5))
model.add(Dense(256, activation='sigmoid'))
model.add(Dropout(0.5))
model.add(Dense(2, activation='softmax'))
opt_adam = tf.keras.optimizers.Adam(lr=0.00001, beta_1=0.9, beta_2=0.999, epsilon=1e-08, decay=0.0)
model.compile(loss='categorical_crossentropy', optimizer=opt_adam, metrics=['accuracy'])
print(model.summary())
输出:
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
conv3d (Conv3D) (None, 16, 224, 2, 3) 32272
_________________________________________________________________
max_pooling3d (MaxPooling3D) (None, 16, 224, 1, 2) 0
_________________________________________________________________
batch_normalization (BatchNo (None, 16, 224, 1, 2) 8
_________________________________________________________________
conv3d_1 (Conv3D) (None, 32, 224, 1, 2) 4640
_________________________________________________________________
max_pooling3d_1 (MaxPooling3 (None, 32, 224, 1, 1) 0
_________________________________________________________________
batch_normalization_1 (Batch (None, 32, 224, 1, 1) 4
_________________________________________________________________
conv3d_2 (Conv3D) (None, 64, 224, 1, 1) 18496
_________________________________________________________________
max_pooling3d_2 (MaxPooling3 (None, 64, 224, 1, 1) 0
_________________________________________________________________
batch_normalization_2 (Batch (None, 64, 224, 1, 1) 4
_________________________________________________________________
flatten (Flatten) (None, 14336) 0
_________________________________________________________________
dropout (Dropout) (None, 14336) 0
_________________________________________________________________
dense (Dense) (None, 256) 3670272
_________________________________________________________________
dropout_1 (Dropout) (None, 256) 0
_________________________________________________________________
dense_1 (Dense) (None, 2) 514
=================================================================
Total params: 3,726,210
Trainable params: 3,726,202
Non-trainable params: 8