소스 검색

Update visualize.py

SunAhong1993 5 년 전
부모
커밋
3b2feefb6e
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      paddlex/cv/models/explanation/visualize.py

+ 1 - 1
paddlex/cv/models/explanation/visualize.py

@@ -28,7 +28,7 @@ def visualize(img_file,
               num_samples=3000, 
               num_samples=3000, 
               batch_size=50,
               batch_size=50,
               save_dir='./'):
               save_dir='./'):
-    model_info = self.get_model_info()
+    model_info = moedl.get_model_info()
     if model_info['status'] != 'Normal':
     if model_info['status'] != 'Normal':
         raise Exception('The explanation only can deal with the Normal model')
         raise Exception('The explanation only can deal with the Normal model')
     model.arrange_transforms(
     model.arrange_transforms(