changed visualize file list path (vis_file_list = cfg.DATASET.VIS_FIL…

…E_LIST)
shijun18 · Oct 23, 2020 · 5c1b896 · 5c1b896
1 parent 20ca269
commit 5c1b896
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pdseg/vis.py b/pdseg/vis.py
@@ -95,7 +95,7 @@ def visualize(cfg,
               local_test=False,
               **kwargs):
     if vis_file_list is None:
-        vis_file_list = cfg.DATASET.TEST_FILE_LIST
+        vis_file_list = cfg.DATASET.VIS_FILE_LIST
     dataset = SegDataset(
         file_list=vis_file_list,
         mode=ModelPhase.VISUAL,