update function name

ivadomed · Mar 11, 2024 · b093d01 · b093d01
1 parent b689e0b
commit b093d01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/init_data_config.py b/scripts/init_data_config.py
@@ -25,7 +25,7 @@ def init_data_config(args):
     file_paths = [os.path.abspath(path.replace('\n', '')) for path in open(args.txt)]
     if args.type == 'LABEL':
         label_paths = file_paths
-        img_paths = [get_img_path_from_label_path(lp) for lp in label_paths]
+        img_paths = [get_img_path_from_mask_path(lp) for lp in label_paths]
         file_paths = label_paths + img_paths
     elif args.type == 'IMAGE':
         img_paths = file_paths