sunyanfang01 5 жил өмнө
parent
commit
0d54ff26d5

+ 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',