TF04——描述卷積計算層
阿新 • • 發佈:2022-03-20
TF04——描述卷積計算層
tf.keras.layers.Conv2D( filters=卷積核個數, kernel_size=卷積核尺寸,#正方形寫核長整數,或(核高h,核寬w) strides=滑動步長,#橫縱向相同寫步長整數,或(縱向步長h,橫向步長w),預設1 padding="same"or"valid",#使用全零填充是"same",不使用是"valid"(預設) activation="relu"or"sigmoid"or"tanh"or"softmax"等,#如有BN此處不寫,如果這一層卷積後還有批標準化,不在這裡進行啟用 input_shape=(高,寬,通道數) # 輸入特徵維度,可省略 )
例項,描述了三層卷積計算,每層用一種表示形式,可以根據自己的習慣選擇一種
model = tf.keras.models.Sequential([ Conv2D(6,5,padding='valid',activation='sigmoid'), MaxPool2D(2,2), Conv2D(6,(5,5),padding='valid',activation='sigmoid'), MaxPool2D(2,(2,2)), Conv2D(filters=6,kernel_size=(5,5),padding='valid',activation='sigmoid'), MaxPool2D(pool_size=(2,2),strides=2) Flatten(), Dense(10,activation='s') ])