lime.py 602 B

12345678910111213141516
  1. import paddlex as pdx
  2. # 下载和解压Imagenet果蔬分类数据集
  3. veg_dataset = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg.tar.gz'
  4. pdx.utils.download_and_decompress(veg_dataset, path='./')
  5. # 下载和解压已训练好的MobileNetV2模型
  6. model_file = 'https://bj.bcebos.com/paddlex/interpret/mini_imagenet_veg_mobilenetv2.tar.gz'
  7. pdx.utils.download_and_decompress(model_file, path='./')
  8. # 加载模型
  9. model = pdx.load_model('mini_imagenet_veg_mobilenetv2')
  10. # 可解释性可视化
  11. pdx.interpret.lime(
  12. 'mini_imagenet_veg/mushroom/n07734744_1106.JPEG', model, save_dir='./')