Skip to Content
移动端AI与ML应用开发:基于iOS和Android
book

移动端AI与ML应用开发:基于iOS和Android

by Laurence Moroney
January 2023
Intermediate to advanced
288 pages
6h 6m
Chinese
China Machine Press
Content preview from 移动端AI与ML应用开发:基于iOS和Android
128
|
8
8.2.1
保存模型
TensorFlow 使用多种不同的模型保存方法,但 TensorFlow 生态系统中最标准化的是
SavedModel 格式。这会将模型保存为
.pb
(用 protobuf)文件格式,作为固定模型的
表示,以及包含其他模型资产或变量的关联目录。这具有将架构与状态分离的决定性优
势,以便以后可以根据需要添加其他状态,或者可以发送模型更新,而无须发送递交任
何模型资产,这些资产本身可能非常大。
要使用这种格式保存,你只需指定输出目录,然后像这样调用
tf.saved_model.save()
export_dir = 'saved_model/1'
tf.saved_model.save(model, export_dir)
你可以在图 8-2 中看到保存的目录结构。
8-2
:保存模型后的目录结构
由于这是一个非常简单的模型,变量文件只有一个分片。较大的模型将拆分为多个,因
此命名为
variables.data-00000-of-00001
。此模型不使用任何资产,因此该文件夹为空。
8.2.2
转换模型
转换模型就像从保存的模型中创建转换器的实例并调用其转换方法一样简单。这将为你
提供 TFLite 格式的模型,然后你可以通过将其字节写入文件流来保存该模型。
这是代码:
converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
tflite_model = converter.convert() ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.

Read now

Unlock full access

More than 5,000 organizations count on O’Reilly

AirBnbBlueOriginElectronic ArtsHomeDepotNasdaqRakutenTata Consultancy Services

QuotationMarkO’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.
Julian F.
Head of Cybersecurity
QuotationMarkI wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.
Addison B.
Field Engineer
QuotationMarkI’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.
Amir M.
Data Platform Tech Lead
QuotationMarkI'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.
Mark W.
Embedded Software Engineer

You might also like

人工智能驱动的商业智能

人工智能驱动的商业智能

Tobias Zwingmann
Lua游戏AI开发指南

Lua游戏AI开发指南

Posts & Telecom Press, David Young
人工智能技术与大数据

人工智能技术与大数据

Posts & Telecom Press, Anand Deshpande, Manish Kumar

Publisher Resources

ISBN: 9787111713081