changed visualize file list path (vis_file_list = cfg.DATASET.VIS_FILE_LIST)

0fa2bf73 · 陈亮 · 52ef0ee5 · 0fa2bf73
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

pdseg/vis.py pdseg/vis.py +1 -1

未找到文件。
--- a/pdseg/vis.py
+++ b/pdseg/vis.py
@@ -93,7 +93,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,