提交 15be4519 编写于 作者: J jiangjiajun

Merge branch 'master' of https://github.com/PaddlePaddle/X2Paddle

### Warning: TensorFlow2Paddle is not stable yet. Only tested on vgg_16/resnet_v1_50/inception_v3 with is_training=False ### Warning: TensorFlow2Paddle is not stable yet. Only tested on vgg_16/resnet_v1_50/inception_v3 with is_training=False
Demo: How to change tensorflow resnet_v1_50 pretrained model to PaddlePaddle model for inference #### Demo: How to change tensorflow resnet_v1_50 pretrained model to PaddlePaddle model for inference
1. Get pretrained_model ##### 1. Get pretrained_model
``` ```
git clone https://github.com/PaddlePaddle/X2Paddle.git git clone https://github.com/PaddlePaddle/X2Paddle.git
...@@ -10,20 +10,20 @@ wget http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz ...@@ -10,20 +10,20 @@ wget http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz
tar xzvf resnet_v2_50_2017_04_14.tar.gz tar xzvf resnet_v2_50_2017_04_14.tar.gz
``` ```
2. Change model to ckpt model with meta file ##### 2. Change model to ckpt model with meta file
``` ```
python demo/save_resnet_ckpt_model.py resnet_v1_50.ckpt ./new_ckpt_model python demo/save_resnet_ckpt_model.py resnet_v1_50.ckpt ./new_ckpt_model
``` ```
3. Export PaddlePaddle model ##### 3. Export PaddlePaddle model
``` ```
python demo/export_resnet_to_paddle_model.py new_ckpt_model/resnet.meta new_ckpt_model fluid_model python demo/export_resnet_to_paddle_model.py new_ckpt_model/resnet.meta new_ckpt_model fluid_model
``` ```
4. Test PaddlePaddle model ##### 4. Test PaddlePaddle model
```python ```python
from fluid_model.mymodel import KitModel from fluid_model.mymodel import KitModel
import paddle.fluid as fluid import paddle.fluid as fluid
...@@ -39,7 +39,7 @@ for f in os.listdir('./fluid_model'): ...@@ -39,7 +39,7 @@ for f in os.listdir('./fluid_model'):
f = f.split('/fluid')[-1] f = f.split('/fluid')[-1]
if f.startswith("param_"): if f.startswith("param_"):
var_list.append(fluid.default_main_program().global_block().var(f)) var_list.append(fluid.default_main_program().global_block().var(f))
fluid.io.load_vars(exe, './fluid', vars=var_list) fluid.io.load_vars(exe, './fluid_model', vars=var_list)
test_data = numpy.random.rand(1, 3, 224, 224) test_data = numpy.random.rand(1, 3, 224, 224)
test_data = numpy.array(test_data, dtype='float32') test_data = numpy.array(test_data, dtype='float32')
...@@ -48,3 +48,6 @@ result = exe.run(fluid.default_main_program(), ...@@ -48,3 +48,6 @@ result = exe.run(fluid.default_main_program(),
fetch_list=[result]) fetch_list=[result])
print(result) print(result)
``` ```
### Link
[MMdnn-Tensorflow](https://github.com/Microsoft/MMdnn/tree/master/mmdnn/conversion/tensorflow)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册