未验证 提交 e33f641b 编写于 作者: T Teng Xi 提交者: GitHub

add Paddle 1.8 PaddleSlim 1.1.1 demo (#304)

* add Paddle 1.8 PaddleSlim 1.1.1 demo
上级 5b92f265
...@@ -2,6 +2,8 @@ ...@@ -2,6 +2,8 @@
本示例将演示如何训练`slimfacenet`及评测`slimfacenet`量化模型。 本示例将演示如何训练`slimfacenet`及评测`slimfacenet`量化模型。
本示例依赖:Paddle 1.8 PaddleSlim 1.1.1
当前示例支持以下人脸识别模型: 当前示例支持以下人脸识别模型:
- `SlimFaceNet_A_x0_60` - `SlimFaceNet_A_x0_60`
......
...@@ -233,9 +233,7 @@ def quant_val_reader_batch(): ...@@ -233,9 +233,7 @@ def quant_val_reader_batch():
test_dataset = LFW(nl, nr) test_dataset = LFW(nl, nr)
test_reader = paddle.fluid.io.batch( test_reader = paddle.fluid.io.batch(
test_dataset.reader, batch_size=1, drop_last=False) test_dataset.reader, batch_size=1, drop_last=False)
shuffle_index = args.seed if args.seed else np.random.randint(1000) shuffle_reader = fluid.io.shuffle(test_reader, 3)
print('shuffle_index: {}'.format(shuffle_index))
shuffle_reader = fluid.io.shuffle(test_reader, shuffle_index)
def _reader(): def _reader():
while True: while True:
...@@ -289,7 +287,6 @@ def main(): ...@@ -289,7 +287,6 @@ def main():
'--start_epoch', default=0, type=int, help='start_epoch') '--start_epoch', default=0, type=int, help='start_epoch')
parser.add_argument( parser.add_argument(
'--total_epoch', default=80, type=int, help='total_epoch') '--total_epoch', default=80, type=int, help='total_epoch')
parser.add_argument('--seed', default=None, type=int, help='shuffle seed')
parser.add_argument( parser.add_argument(
'--save_frequency', default=1, type=int, help='save_frequency') '--save_frequency', default=1, type=int, help='save_frequency')
parser.add_argument( parser.add_argument(
...@@ -336,8 +333,10 @@ def main(): ...@@ -336,8 +333,10 @@ def main():
sample_generator=quant_val_reader_batch(), sample_generator=quant_val_reader_batch(),
model_filename=None, #'model', model_filename=None, #'model',
params_filename=None, #'params', params_filename=None, #'params',
batch_size=100, save_model_filename=None, #'model',
batch_nums=10) save_params_filename=None, #'params',
batch_size=np.random.randint(80, 160),
batch_nums=np.random.randint(4, 10))
elif args.action == 'test': elif args.action == 'test':
[inference_program, feed_target_names, [inference_program, feed_target_names,
fetch_targets] = fluid.io.load_inference_model( fetch_targets] = fluid.io.load_inference_model(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册