From 33cac039975f1561b1fbc03ef3c19cf4def5a8d9 Mon Sep 17 00:00:00 2001 From: huzhiqiang <912790387@qq.com> Date: Tue, 26 May 2020 13:01:39 +0800 Subject: [PATCH] [Python Demo] Modify python demo to improve its usability --- lite/demo/python/mobilenetv1_full_api.py | 7 ++----- lite/demo/python/mobilenetv1_light_api.py | 4 +--- 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/lite/demo/python/mobilenetv1_full_api.py b/lite/demo/python/mobilenetv1_full_api.py index c3a6bd077b..7d713dfeea 100644 --- a/lite/demo/python/mobilenetv1_full_api.py +++ b/lite/demo/python/mobilenetv1_full_api.py @@ -20,9 +20,6 @@ from __future__ import division from __future__ import print_function import argparse -import sys -sys.path.append('../../python/lib') - from paddlelite.lite import * # Command arguments @@ -42,8 +39,8 @@ def RunModel(args): config.set_param_file(args.param_file) else: config.set_model_dir(args.model_dir) - # For x86, you can set places = [Place(TargetType.X86, PrecisionType.FP32)] - places = [Place(TargetType.ARM, PrecisionType.FP32)] + # For arm platform (armlinux), you can set places = [Place(TargetType.ARM, PrecisionType.FP32)] + places = [Place(TargetType.X86, PrecisionType.FP32)] config.set_valid_places(places) # 2. Create paddle predictor diff --git a/lite/demo/python/mobilenetv1_light_api.py b/lite/demo/python/mobilenetv1_light_api.py index 5847c78193..a3e2db6c0d 100644 --- a/lite/demo/python/mobilenetv1_light_api.py +++ b/lite/demo/python/mobilenetv1_light_api.py @@ -20,8 +20,6 @@ from __future__ import division from __future__ import print_function import argparse -import sys -sys.path.append('../../python/lib') from paddlelite.lite import * @@ -33,7 +31,7 @@ parser.add_argument( def RunModel(args): # 1. Set config information config = MobileConfig() - config.set_model_dir(args.model_dir) + config.set_model_from_file(args.model_dir) # 2. Create paddle predictor predictor = create_paddle_predictor(config) -- GitLab