diff --git a/TensorFlow2Paddle/README.md b/TensorFlow2Paddle/README.md index 13fda93870ba851e5138383d3d9f71cc2aca9529..6c90c608ba0a90cde0e76f75b25e1b6d340956ed 100644 --- a/TensorFlow2Paddle/README.md +++ b/TensorFlow2Paddle/README.md @@ -1,7 +1,7 @@ ### 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]