Merge pull request #373 from chliang/release/v0.6.0

f37a862a · wuzewu · GitHub · 52ef0ee5 · 0fa2bf73 · f37a862a
显示空白变更内容
内联并排

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,