small bug fix

Former-commit-id: 7a27ee3

.gitignore

@@ -201,3 +201,5 @@ examples/process_one.py
 
 tests/temp_folder*
 examples/data_test
+data
+demo.py

imgtools/autopipeline.py

@@ -186,6 +186,7 @@ def save_data(self):
         files = glob.glob(os.path.join(self.output_directory,".temp","*.pkl"))
         for file in files:
             subject_id = ("_").join(file.replace("/","_").replace(".","_").split("_")[-3:-1])
+            subject_id = file.replace(".","/").split("/")[-2][5:] #From temp_ considers
             with open(file,"rb") as f:
                 metadata = pickle.load(f)
             self.output_df.loc[subject_id, list(metadata.keys())] = list(metadata.values())

imgtools/io/dataset.py

@@ -72,10 +72,10 @@ def load_from_nrrd(
                 path_mod = os.path.join(path,extension.split(".")[0],f"{subject_id}_{extra}{extension}.nrrd")
                 #All modalities except RTSTRUCT should be of type torchIO.ScalarImage
                 if os.path.exists(path_mod):
-                    if col!="RTSTRUCT":
+                    if col.split("_")[0]!="RTSTRUCT":
                         temp[f"mod_{col}"] = tio.ScalarImage(path_mod)
                     else:
-                        temp[f"mod_{col}"] = tio.LabelImage(path_mod)
+                        temp[f"mod_{col}"] = tio.LabelMap(path_mod)
                 else:
                     temp[f"mod_{col}"] = None
                 #For including metadata