提交 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
Demo: How to change tensorflow resnet_v1_50 pretrained model to PaddlePaddle model for inference
1. Get pretrained_model
#### Demo: How to change tensorflow resnet_v1_50 pretrained model to PaddlePaddle model for inference
##### 1. Get pretrained_model
```
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
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
```
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
```
4. Test PaddlePaddle model
##### 4. Test PaddlePaddle model
```python
from fluid_model.mymodel import KitModel
import paddle.fluid as fluid
......@@ -39,7 +39,7 @@ for f in os.listdir('./fluid_model'):
f = f.split('/fluid')[-1]
if f.startswith("param_"):
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.array(test_data, dtype='float32')
......@@ -48,3 +48,6 @@ result = exe.run(fluid.default_main_program(),
fetch_list=[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.
先完成此消息的编辑!
想要评论请 注册