目录
python 保存模型、加载模型 Joblib
支持数据类型:字典,列表,numpy。
保存模型:
saved_data = {"ROI": pts}
joblib.dump(value=saved_data, filename="config.pkl")
加载模型:
model1 = joblib.load(filename="config.pkl")
print(model1)
numpy保存与加载数据:
np.save('test_a.npy', image)
a = np.load('test_a.npy')
diff=abs(a-image)
pytorch 保存与加载
# 保存模型
torch.save(model, 'model.pth\pkl\pt') #一般形式torch.save(net, PATH)
# 加载模型
model = torch.load('model.pth\pkl\pt') #一般形式为model_dict=torch.load(PATH)