sunyanfang01 5 жил өмнө
parent
commit
084264488b

+ 0 - 35
tutorials/interpret/interpret.py

@@ -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='./')