|
@@ -4,7 +4,6 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
|
|
|
|
|
|
|
import os.path as osp
|
|
import os.path as osp
|
|
|
import paddlex as pdx
|
|
import paddlex as pdx
|
|
|
-from paddlex.cls import transforms
|
|
|
|
|
|
|
|
|
|
# 下载和解压Imagenet果蔬分类数据集
|
|
# 下载和解压Imagenet果蔬分类数据集
|
|
|
veg_dataset = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg.tar.gz'
|
|
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')
|
|
model = pdx.load_model('mini_imagenet_veg_mobilenetv2')
|
|
|
|
|
|
|
|
-
|
|
|
|
|
# 定义测试所用的数据集
|
|
# 定义测试所用的数据集
|
|
|
test_dataset = pdx.datasets.ImageNet(
|
|
test_dataset = pdx.datasets.ImageNet(
|
|
|
data_dir='mini_imagenet_veg',
|
|
data_dir='mini_imagenet_veg',
|
|
@@ -37,4 +35,4 @@ pdx.interpret.visualize(
|
|
|
model,
|
|
model,
|
|
|
test_dataset,
|
|
test_dataset,
|
|
|
algo='normlime',
|
|
algo='normlime',
|
|
|
- save_dir='./')
|
|
|
|
|
|
|
+ save_dir='./')
|