前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Tensorflow1.X迁移到2.0教程

Tensorflow1.X迁移到2.0教程

作者头像
CristianoC
发布2020-06-02 11:28:56
6420
发布2020-06-02 11:28:56
举报

最近发现tf升级2.0之后有些api更新了,所以代码就无法直接迁移,所以做一下记录。

目录

  • 单脚本转换
  • 文件夹批量转换
  • 注意事项

单脚本转换

  • 先进入到你要转换脚本(eg:yolov3.py)的目录,然后:
代码语言:javascript
复制
tf_upgrade_v2 --infile yolov3.py --outfile yolov3.py

文件夹批量转换

  • 先进入你要转换的文件夹根目录(eg:yolov3/),然后可以转换文件下所有py文件:
代码语言:javascript
复制
tf_upgrade_v2 --intree yolov3/ --outtree yolov3/ --copyotherfiles False

注意事项

1.转换好之后可能会报类似于“tf.placeholder() is not compatible with eager execution”这样的错,只需要在正常import tensorflow后面加上这一句:

代码语言:javascript
复制
import tensorflow as tf
tf.compat.v1.disable_eager_execution()

2.tf2.0版本弃用了类似于tf.flags这样的库,可能要重新装absl.flags或切换至 tensorflow/addons 中的软件包来成功迁移代码。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-02-24,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 计算机视觉漫谈 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 目录
  • 单脚本转换
  • 文件夹批量转换
  • 注意事项
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档