基于TensorFlow的目标检测模型使用与代码实现
1. 模型下载
要从Google Drive下载模型,需登录Google Drive,将训练好的模型下载到本地计算机,应下载整个 final_model 目录。
2. 在TensorBoard中可视化训练结果
若要查看训练统计信息和模型结果,可在Colab中使用以下代码启动TensorBoard仪表盘:
%load_ext tensorboard
%tensorboard --logdir /content/computer_vision/pet_detection_model
第一行代码加载TensorBoard笔记本扩展,这会将TensorBoard仪表盘嵌入到Colab屏幕中显示。
若想在本地计算机而非Colab上离线评估模型,需下载保存模型检查点的整个 pet_detection_model 目录,因为导出训练模型的 final_model 目录不包含完整的模型统计信息和训练结果。在计算机终端(或命令提示符)中,通过传递 pet_detection_model 目录的路径来启动TensorBoard,确保处于虚拟环境中,命令如下:
(cv) username$ tensorboard --logdir ~/Downloads/pet_detection_model
</
超级会员免费看
订阅专栏 解锁全文
35

被折叠的 条评论
为什么被折叠?



