Android APP 集成 Unet进行图像语义分割【tensorflow】

环境：

WIN7 64 + Android Studio3.3

Python 3.6

Keras 2.3

TF 1.9

概述：

1.先用Keras训练网络，保存为h5文件【model.save('xxx.h5')】

2.用Keras2pb.py 把h5文件转成Tf的pb文件

3.在Android src/main下新建Assets文件夹，把pb放到里面

4.Android gradle【app】里implementation一个Tf包

implementation 'org.tensorflow:tensorflow-android:+'

5.调用TF JAVA接口feed【input】 run fetch【output】

     

 Keras转PB：

# coding=utf-8
import sys

from keras.models import load_model
import tensorflow as tf
import os
import os.path as osp
from keras import backend as K


def freeze_session(session, keep_var_names=None, output_names=None, clear_devices=True):
    """
    Freezes the state of a session into a prunned computation graph.

    Creates a new computation graph where variabl