【发布时间】:2025-12-21 01:30:10
【问题描述】:
我有一个微调版本的 inceptionV3 模型,我想在新数据集上进行测试。但是,我收到错误No model found in config file.
这是我的代码,
from tensorflow import keras
model = keras.models.load_model('/home/saved_model/CNN_inceptionv3.h5')
CLASS_1_data = '/home/spectrograms/data/c1'
def label_img(img):
word_label = img[:5]
if img[1] == '1':
return [1,0]
elif img[1] == '3':
return [0,1]
def create_data(data,loc): #loads data into a list
for img in tqdm(os.listdir(loc)):
label = label_img(img)
path = os.path.join(loc,img)
img = Image.open(path)
img = ImageOps.grayscale(img)
# w,h = img.size
# img = img.resize((w//3,h//3))
data.append([np.array(img),np.array(label)])
return data
def make_X_and_Y(set): #split data into numpy arrays of inputs and outputs
set_X,set_Y = [],[]
n = len(set)
for i in range(n):
set_X.append(set[i][0])
set_Y.append(set[i][1])
return np.array(set_X),np.array(set_Y)
data = []
data = create_data(data,CLASS_1_data)
data = np.array(data)
X_data,Y_data = make_X_and_Y(data)
X_data = X_data.astype('float32')
X_data /= 255
results = model.evaluate(X-data, Y_data, batch_size=5)
这里的错误是什么?如何更正它并测试我的模型?
【问题讨论】:
-
你是只保存权重还是只保存模型? model.save 还是 model.save_weights?
-
我使用了model.save_weights
标签: tensorflow machine-learning keras deep-learning conv-neural-network