diff --git a/tutorials/interpret/interpret.py b/tutorials/interpret/interpret.py deleted file mode 100644 index c90870da2864a52f8ce41f9575d821b3d21994eb..0000000000000000000000000000000000000000 --- a/tutorials/interpret/interpret.py +++ /dev/null @@ -1,35 +0,0 @@ -import os -# 选择使用0号卡 -os.environ['CUDA_VISIBLE_DEVICES'] = '0' - -import os.path as osp -import paddlex as pdx - -# 下载和解压Imagenet果蔬分类数据集 -veg_dataset = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg.tar.gz' -pdx.utils.download_and_decompress(veg_dataset, path='./') - -# 下载和解压已训练好的MobileNetV2模型 -model_file = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg_mobilenetv2.tar.gz' -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', - file_list=osp.join('mini_imagenet_veg', 'test_list.txt'), - label_list=osp.join('mini_imagenet_veg', 'labels.txt'), - transforms=model.test_transforms) - -# 可解释性可视化 -pdx.interpret.lime( - 'mini_imagenet_veg/mushroom/n07734744_1106.JPEG', - model, - save_dir='./') -pdx.interpret.normlime( - 'mini_imagenet_veg/mushroom/n07734744_1106.JPEG', - model, - test_dataset, - save_dir='./')