Update eval_depth_pp.py

ArminMasoumian · Feb 28, 2022 · e2de614 · e2de614
1 parent b34081b
commit e2de614
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/eval_depth_pp.py b/scripts/eval_depth_pp.py
@@ -122,5 +122,5 @@ def evaluate(MODEL_PATH, CFG_PATH, GT_PATH):
 if __name__ == "__main__":
     CFG_PATH = '../config/cfg_kitti_fm.py'#path to cfg file
     GT_PATH = '/media/sconly/harddisk/data/kitti/kitti_raw/rawdata/gt_depths.npz'#path to kitti gt depth
-    MODEL_PATH = '/media/sconly/harddisk/weight/fm_depth.pth'#path to model weights
-    evaluate(MODEL_PATH, CFG_PATH, GT_PATH)
+    MODEL_PATH = '/media/sconly/harddisk/weight/epoch_20.pth'#path to model weights
+    evaluate(MODEL_PATH, CFG_PATH, GT_PATH)