diff --git a/tutorials/interpret/interpret.py b/tutorials/interpret/interpret.py index dc107405fa9d442d941962387258ea96de94afdf..f52d1053f5dcb1b2f1a585f50e9e0f2b1cb13ef2 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='./')