如何在 Tensorflow 2 中使用腌制的預訓練權重？
[英]How to use pickled pretrained weights in Tensorflow 2?
問題描述
我正在嘗試加載在 tf2 中使用 tensorflow 1.x 訓練的 model 中生成的 pkl 文件。 model.load_weights(checkpoint_path)似乎不起作用。 有什么建議么？
1 個解決方案
解決方案1
0 已采納 2021-05-06 20:19:38
嘗試加載泡菜文件然后獲取權重。
import pickle
import tensorflow as tf
model = pickle.loads('<path>')
loaded_model = model.get_weights()
Keras：在theano和tensorflow之間轉換預訓練的權重
[英]Keras: convert pretrained weights between theano and tensorflow
將 CNN Pytorch 中的預訓練權重傳遞給 Tensorflow 中的 CNN
[英]Pass pretrained weights in CNN Pytorch to a CNN in Tensorflow
如何使用Tensorflow的實例分割預訓練的MaskRCNN模型？
[英]How to use Instance segmentation pretrained MaskRCNN model by Tensorflow?
僅用於人/車輛檢測的預訓練 CNN（tensorflow/darknet/caffe）權重
[英]Pretrained CNN(tensorflow/darknet/caffe) weights for human/vehicle detection only
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.
相關問題
Tensorflow：如何在新圖中使用預訓練權重？
如何在Tensorflow中加載預訓練的LSTM模型權重
如何使用預訓練模型的 model 架構但沒有權重
Keras：在theano和tensorflow之間轉換預訓練的權重
將 CNN Pytorch 中的預訓練權重傳遞給 Tensorflow 中的 CNN
如何在Tensorflow中使用事件重量？
Tensorflow：使用預訓練的初始模型
如何使用Tensorflow的實例分割預訓練的MaskRCNN模型？
如何在Tensorflow中使用預訓練的Word2Vec模型
僅用於人/車輛檢測的預訓練 CNN（tensorflow/darknet/caffe）權重
相關標簽