From 7e9a2249007068539aadf7352a4d181401812969 Mon Sep 17 00:00:00 2001 From: Jason Date: Mon, 28 Jan 2019 10:45:58 +0800 Subject: [PATCH] Update README.md --- tensorflow2fluid/README.md | 25 ++++++++++++++++++++----- 1 file changed, 20 insertions(+), 5 deletions(-) diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 2754d3a..7b2ab8c 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -19,11 +19,26 @@ my_model.py|基于PaddlePaddle实现的模型网络结构python代码 ref_name.txt|my_model.py中各tensor与原TensorFlow模型中的tensor对应关系 const_\*/params_\*|转换后的模型参数文件 -tensorflow2fluid在模型转换过程中，以tensorflow计算图中的节点为粒度，遍历图中的节点，并将每个节点所对应的OP转换为基于PaddlePaddle实现的python网络结构代码，目前支持OP如下表所示 - -TensorFlow OP名|Python接口 -:---------------------------:|:------------------------------------------:| -conv2d|tf.layers.conv2d +tensorflow2fluid在模型转换过程中，以tensorflow计算图中的节点为粒度，遍历图中的节点，并将每个节点所对应的OP转换为基于PaddlePaddle实现的python网络结构代码，目前支持OP如下表所示，需要注意的是，并非所有OP都需要转成PaddlePaddle对应的代码实现，如Identity，switch等OP，在实际转换过程中，都直接将输出表示为输入即可 + +| TensorFlow OP | Python Api | TensorFlow OP | Python Api | +| ------------------- | ---------- | ---------------------- | ---------- | +| VariableV2 | 1 | placeholderwithdefault | 17 | +| Identity | 2 | switch | 18 | +| Placeholder | 3 | merge | 19 | +| Const | 4 | MaxPool | 20 | +| Conv2D | 5 | Squeeze | 21 | +| BiasAdd | 6 | Add | 22 | +| Relu | 7 | Mean | 23 | +| Conv2dBackpropInput | 8 | DepthwiseConv2dNative | 24 | +| FusedBatchNorm | 9 | Pad | 25 | +| ConcatV2 | 10 | StridedSlice | 26 | +| AvgPool | 11 | ResizeNearestNeighbor | 27 | +| Rsqrt | 12 | Maximum | 28 | +| Mul | 13 | Minimum | 9 | +| Sub | 14 | Sigmoid | 30 | +| Shape | 15 | Pack | 31 | +| Reshape | 16 | | | 模型中所使用的代码，一般而言并不能直接能过模型训练时所使用的tensorflow代码中就能完全看出来。比如在python模型代码中所使用到的`tf.contrib.layers.fully_connected`就涉及到如下OP -- GitLab