【发布时间】:2021-09-10 04:25:57
【问题描述】:
def get_model():
gmodel=Sequential([
Conv2D(64, (3, 3), activation='relu', input_shape=(256,256,3)),
Conv2D(64,(3,3),activation='relu'),
MaxPooling2D((1,1),strides=2,padding='same'),
Conv2D(128, (3, 3), activation='relu'),
Conv2D(128, (3, 3), activation='relu'),
MaxPooling2D((1,1), strides=2, padding='same'),
Conv2D(256, (3, 3), activation='relu'),
Conv2D(256, (3, 3), activation='relu'),
MaxPooling2D((1,1), strides=2, padding='same'),
Conv2D(512, (3, 3), activation='relu'),
Conv2D(512, (3, 3), activation='relu'),
MaxPooling2D((1,1), strides=2, padding='same'),
Conv2D(512, (3, 3), activation='relu'),
Conv2D(512, (3, 3), activation='relu'),
MaxPooling2D((1,1), strides=2, padding='same'),
Flatten(),
Dense(4, activation='softmax')
])
return gmodel
model=get_model()
model.compile(
optimizer=tf.keras.optimizers.Adam(0.0001),
loss="sparse_categorical_crossentropy",
metrics=['accuracy']
)
history=model.fit(train,epochs=5)
我想使用 VGG 模型将图片分类为 4 类,但准确率始终在 25% 左右。我检查了图片和标签,它们都是正确的,我也尝试了其他一些模型和优化器,但准确性也没有太大的提高。我该怎么办。我的 GPU 是 2070,tensorflow 是 2.0,python 是 3.6
【问题讨论】:
标签: machine-learning deep-learning computer-vision tensorflow2.0 vgg-net