keras保存模型的两种方式:
model.save("m1.h5") # 既保存了模型的图结构,又保存了模型的参数
model.save_weights("m2.h5") #只保存了模型的参数
keras加载模型的两种方式:
#第一种(通用性网络可用)
from keras.models import load_model
model = load_model("weight.h5")
#第二种(适用于有创新的网络,例如自创的激活函数等)
from Nets.mobilenetV3_small import MobileNetv3_small
model = MobileNetv3_small((224,224,3),2)
model.load_weight("weight.h5")