change img path

lyp2333 · Jul 30, 2019 · 409e21e · 409e21e
1 parent dca1afe
commit 409e21e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/Code/4_viewer/6_hook_for_grad_cam.py b/Code/4_viewer/6_hook_for_grad_cam.py
@@ -130,10 +130,9 @@ def gen_cam(feature_map, grads):
 if __name__ == '__main__':
 
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-    path_img = os.path.join(BASE_DIR, "../../Data/cam_img/", "test_img_8.png")
-    path_img = "/Users/tingsongyu/Desktop/t.png"
-    path_net = os.path.join(BASE_DIR, "../../Data/", "net_params_72p.pkl")
-    output_dir = os.path.join(BASE_DIR, "../../Result/backward_hook_cam/")
+    path_img = os.path.join(BASE_DIR, "..", "..", "Data/cam_img/", "test_img_8.png")
+    path_net = os.path.join(BASE_DIR, "..", "..", "Data/", "net_params_72p.pkl")
+    output_dir = os.path.join(BASE_DIR, "..", "..", "Result/backward_hook_cam/")
 
     classes = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
     fmap_block = list()