专栏首页计算机视觉漫谈Tensorflow1.X迁移到2.0教程

Tensorflow1.X迁移到2.0教程

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

目录

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

单脚本转换

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

文件夹批量转换

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

注意事项

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

import tensorflow as tf
tf.compat.v1.disable_eager_execution()

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

本文分享自微信公众号 - 计算机视觉漫谈(gh_41e38694bc19),作者:CristianoC

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2020-02-24

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 从零开始手把手带你了解并复现AlexNet

    概述论文之前笔者啰嗦多几句,一些初学者可能喜欢上网看一些网上大牛讲解的一些网络结构进行学习,不是说这样不好,一些大牛会把一些晦涩的理论知识讲解的比较生动,但一些...

    CristianoC
  • 卷积神经网络(CNN)

    假设给定一张图(可能是字母X或者字母O),通过CNN即可识别出是X还是O,如下图所示

    CristianoC
  • 目标检测之YOLO v3-You Only Look Once(三)

    进一步加大对小框的损失,值的范围是,边界框的尺寸越小,bbox_loss_scale 的值就越大。box_loss_scale可以弱化边界框尺寸对损失值的影响;

    CristianoC
  • TensorFlow中常量与变量的基本操作演示

    TensorFlow中常量与变量的基本操作演示 本文将介绍TensorFlow中的基本算法运算与矩阵运算,介绍Tensorflow中常量、变量、操作符等基本运算...

    OpenCV学堂
  • 【干货】Batch Normalization: 如何更快地训练深度神经网络

    【导读】本文是谷歌机器学习工程师 Chris Rawles 撰写的一篇技术博文,探讨了如何在 TensorFlow 和 tf.keras 上利用 Batch N...

    WZEARW
  • tf24: GANs—生成明星脸

    GANs是Generative Adversarial Networks的简写,中文翻译为生成对抗网络,它最早出现在2014年Goodfellow发表的论文中...

    MachineLP
  • 一看就懂的Tensorflow实战(模型的保存与读取)

    首先,我们从一个直观的例子,讲解如何实现Tensorflow模型参数的保存以及保存后模型的读取。 然后,我们在之前多层感知机的基础上进行模型的参数保存,以及参数...

    AI异构
  • ubuntu 下增加swap

    超级大猪
  • tensorflow之tf.tile\tf.slice等函数的基本用法解读

    解读: tensorflow中的tile()函数是用来对张量(Tensor)进行扩展的,其特点是对当前张量内的数据进行一定规则的复制。最终的输出张量维度不变。

    学到老
  • 机器学习入门之HelloWorld(Tensorflow)

    1 环境搭建 (Windows) 安装虚拟环境 Anaconda,方便python包管理和环境隔离。 Anaconda3 4.2 http://mirrors...

    腾讯Bugly

扫码关注云+社区

领取腾讯云代金券