#import all the libraries
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix,classification_report
import os
import sklearn.utils
path='/content/drive/My Drive/Drug/spheroid_imageset/untreated/training_set'
X=np.load(path+'/X.npy')
Y=np.load(path+'/Y.npy')
X=X.reshape(-1,224,224,1)
X,Y = sklearn.utils.shuffle(X,Y)
tf.keras.callbacks.EarlyStopping(
# monitor="val_loss",
# min_delta=0,
patience=3,
verbose=0,
mode="auto",
baseline=None,
restore_best_weights=False,
)
image_size=224
base_model = tf.keras.applications.MobileNet(weights=None,include_top=False,input_shape=(image_size,image_size,1))
base_model.trainable = False
x = base_model.output
x = tf.keras.layers.GlobalAveragePooling2D()(x)
x = tf.keras.layers.Dropout(0.50)(x)
x = tf.keras.layers.Dense(16,activation='relu')(x)
x = tf.keras.layers.Dense(3,activation="softmax")(x)
model = tf.keras.Model(inputs=base_model.input, outputs=x)
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'])
history=model.fit(X,Y,epochs=50,validation_split=0.25,callbacks=[callback])发布于 2021-01-01 14:47:03
权重:“无”(随机初始化)、“ImageNet”(关于ImageNet的预培训)或要加载的权重文件的路径。默认为imagenet。
使用weights=None,它被随机初始化,并且
使用base_model.trainable = False,它没有被训练。
基本上是你的长官。图层没有接受训练。
https://datascience.stackexchange.com/questions/87392
复制相似问题