DL-部署
城俊BLOG
从此好好码代码。。
展开
-
模型转换:tensorflow2 模型转pb,pb转tflite
def img_to_batch(): img = cv2.imread('xxxx.jpg') img = cv2.resize(img, (112,112)).astype(np.float32) / 255. img_batch = np.expand_dims(img,0) def tf2_to_pb(): # load tf2 model = AAA(xxx) ckpt_path = tf.train.latest_checkpoint('./xx原创 2021-08-13 09:57:45 · 1326 阅读 · 2 评论 -
模型转换:pytorch模型转onnx, onnx转tensorflow, tensorflow转tflite
pass原创 2021-08-19 19:56:08 · 3954 阅读 · 9 评论