From f7d2a00ef914833b69808d50d67b027436f421dd Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Tue, 19 May 2020 20:17:32 +0800 Subject: [PATCH] add interpret --- tutorials/interpret/interpret.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tutorials/interpret/interpret.py b/tutorials/interpret/interpret.py index dc10740..f52d105 100644 --- a/tutorials/interpret/interpret.py +++ b/tutorials/interpret/interpret.py @@ -4,7 +4,6 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '0' import os.path as osp import paddlex as pdx -from paddlex.cls import transforms # 下载和解压Imagenet果蔬分类数据集 veg_dataset = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg.tar.gz' @@ -17,7 +16,6 @@ pdx.utils.download_and_decompress(model_file, path='./') # 加载模型 model = pdx.load_model('mini_imagenet_veg_mobilenetv2') - # 定义测试所用的数据集 test_dataset = pdx.datasets.ImageNet( data_dir='mini_imagenet_veg', @@ -37,4 +35,4 @@ pdx.interpret.visualize( model, test_dataset, algo='normlime', - save_dir='./') \ No newline at end of file + save_dir='./') -- GitLab