Browse Source

fix the tutorial

sunyanfang01 5 năm trước cách đây
mục cha
commit
0d54ff26d5
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      tutorials/interpret/interpret.py

+ 2 - 2
tutorials/interpret/interpret.py

@@ -34,12 +34,12 @@ model = pdx.load_model('mini_imagenet_veg_mobilenetv2')
 save_dir = 'interpret_results'
 if not osp.exists(save_dir):
     os.makedirs(save_dir)
-visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
+pdx.interpret.visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
           model,
           test_dataset, 
           algo='lime',
           save_dir=save_dir)
-visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
+pdx.interpret.visualize('mini_imagenet_veg/mushroom/n07734744_1106.JPEG', 
           model, 
           test_dataset, 
           algo='normlime',