From 0ca4cd5b88854f93401f5835d8519265d07ce27e Mon Sep 17 00:00:00 2001
From: SunAhong1993 <48579383+SunAhong1993@users.noreply.github.com>
Date: Sat, 16 May 2020 16:12:55 +0800
Subject: [PATCH] Update visualize.py

---
 paddlex/cv/models/explanation/visualize.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddlex/cv/models/explanation/visualize.py b/paddlex/cv/models/explanation/visualize.py
index bd097cf..c37b2e8 100644
--- a/paddlex/cv/models/explanation/visualize.py
+++ b/paddlex/cv/models/explanation/visualize.py
@@ -28,6 +28,9 @@ def visualize(img_file,
               num_samples=3000, 
               batch_size=50,
               save_dir='./'):
+    model_info = self.get_model_info()
+    if model_info['status'] != 'Normal':
+        raise Exception('The explanation only can deal with the Normal model')
     model.arrange_transforms(
                 transforms=model.test_transforms, mode='test')
     tmp_transforms = copy.deepcopy(model.test_transforms)
@@ -110,4 +113,4 @@ def precompute_for_normlime(predict_func, normlime_dataset, num_samples=3000, ba
             save_dir=save_dir)
 
 
-    
\ No newline at end of file
+    
-- 
GitLab