1. 程式人生 > 其它 >TF04——描述卷積計算層

TF04——描述卷積計算層

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')
])