Tensorflow：Android调用Tensorflow Mobile版本API（2）-基于Android的调用

最新推荐文章于 2023-04-16 21:50:39 发布

胡大炮的妖孽人生

最新推荐文章于 2023-04-16 21:50:39 发布

阅读量897

点赞数 1

分类专栏： android python3

本文链接：https://blog.csdn.net/huplion/article/details/79302770

版权

本文介绍如何在Android项目中导入1.6.0版本的TensorFlow Mobile库，详细步骤包括创建新项目、添加离线库依赖、放置模型文件及封装TensorFlow Mobile API。通过这些操作，可以在Android应用中顺利调用TensorFlow模型。项目源码已上传至GitHub，供读者参考。

摘要由CSDN通过智能技术生成

对上一篇博客中代码略做修改，在训练完成之后进行模型导出操作

# y = x^2 + 1

import tensorflow as tf
import numpy as np
import random

def get_batch(size=128):
    xs = []
    ys = []
    for i in range(size):
        x = random.random() * 2
        y = x * x + 1
        xs.append(x)
        ys.append(y)
    return np.array(xs), np.array(ys)




X = tf.placeholder(tf.float32, [None,1], name='input')
Y = tf.placeholder(tf.float32, [None,1])
def my_dnn():
    x = tf.reshape(X, shape=[-1, 1])
    w1 = tf.Variable(tf.random_normal(shape=[1,256], mean=0.0,