TensorFlow
文章平均质量分 66
qiumokucao
这个作者很懒,什么都没留下…
展开
-
pytorch模型转到TensorFlow lite:pytorch->onnx->tensorflow->tensorflow lite
现在很多算法都是用pytorch框架训练的,但是在移动端部署很多又使用TensorFlow lite,因此需要将pytorch模型转换到TensorFlow lite。将pytorch模型转到TensorFlow lite的流程是pytorch->onnx->tensorflow->tensorflow lite,本文记录一下踩坑的过程。1、pytorch转onnx这一步比较简单,使用pytorch自带接口就行。不过有一点需要注意的,就是opset版本,可能会影响后续的转换。原创 2021-05-14 17:45:27 · 5390 阅读 · 2 评论 -
把Python变量记录到TensorFlow的summary中
在用tensorboard可视化时,遇到过一个坑:打算每训练一定步数,将验证集准确率记录下来。一般做法如下:...# 1、计算准确率correct_prediction = tf.equal(tf.argmax(logits, 1), tf.argmax(y, 1))accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.f...原创 2018-08-07 22:27:10 · 1902 阅读 · 0 评论 -
TensorFlow训练模型的过程中打开tensorboard
在训练的过程中,想通过tensorboard实时观察训练损失和验证集准确率,一直出错,打开tensorboard后在浏览器查看,然后训练就停止了,提示信息如下:File "D:/ProgramData/PycharmProjects/tf_learn/mnist/mnist_train.py", line 88, in main train() File "D:/ProgramD...原创 2018-08-08 10:46:08 · 4395 阅读 · 1 评论 -
TensorFlow学习实践(一):使用TFRecord格式数据和队列进行模型训练和预测
本文以mnist为例,介绍如何使用TFRecord格式数据和队列进行模型训练和预测。参考:1、cifar102、https://tensorflow.google.cn/guide/datasetsTFRecord格式数据的制作参见将mnist数据转成原始图片数据再转成TFRecord格式目录一、输入数据的解析和预处理二、定义模型三、计算损失并定义训练操作四、模...原创 2018-08-26 19:26:11 · 2904 阅读 · 0 评论 -
TensorFlow学习实践(二):使用TFRecord格式数据和tf.data API进行模型训练和预测
本文以mnist为例,介绍如何使用TFRecord格式数据和tf.data API进行模型训练和预测。参考:1、cifar102、https://tensorflow.google.cn/guide/datasets目录一、数据解析二、定义模型、损失和训练操作三、模型训练一、数据解析def parse_data(example_proto): feat...原创 2018-08-26 20:10:58 · 2646 阅读 · 2 评论 -
将mnist数据转成原始图片数据再转成TFRecord格式
1、将mnist数据转成原始图片数据def convert_mnist_img(data, save_path): for i in range(data.images.shape[0]): img = data.images[i].reshape([28, 28, 1]) img = (img * 255).astype(np.uint8) ...原创 2018-08-20 14:49:16 · 763 阅读 · 0 评论 -
TensorFlow学习实践(四):使用TFRecord格式数据和tf.contrib.slim API进行模型训练和预测
本文以mnist为例,介绍如何使用TFRecord格式数据和tf.contrib.slim API进行模型训练和预测。参考:1、https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/slim目录一、数据输入二、模型定义三、模型训练四、模型验证一、数据输入数据输入与上篇中的相...原创 2018-08-30 21:55:54 · 1037 阅读 · 0 评论 -
TensorFlow学习实践(三):使用TFRecord格式数据和tf.estimator API进行模型训练和预测
本文以mnist为例,介绍如何使用TFRecord格式数据和tf.estimator API进行模型训练和预测。参考:1、https://tensorflow.google.cn/tutorials/estimators/cnn目录一、数据输入二、模型定义三、模型训练和验证一、数据输入def input_fn(filenames, training): d...原创 2018-08-29 21:58:43 · 3931 阅读 · 1 评论 -
TensorFlow学习实践(五):基于vgg-16、inception_v3、resnet_v1_50模型进行fine-tune全过程
本文基于vgg-16、inception_v3、resnet_v1_50模型进行fine-tune,完成一个二分类模型的训练。目录一、环境准备二、准备数据三、数据解析及图片预处理四、模型定义五、模型训练六、模型预测最后:完整代码 一、环境准备我使用了TensorFlow的model库中的slim模块,路径:https://github.com/tenso...原创 2018-09-11 10:49:55 · 4859 阅读 · 8 评论