Fix and add subset filenames

gzuidhof · Jun 20, 2016 · a6e9819 · a6e9819
1 parent 3a7cc0d
commit a6e9819
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/config/subset_to_filenames.pkl b/config/subset_to_filenames.pkl
diff --git a/src/subset.py b/src/subset.py
@@ -9,7 +9,7 @@ def make_filename_to_subset_dict():
     for subset in range(10):
         files = glob.glob(in_folder+'subset{0}'.format(subset)+'/*.mhd')
         print "Subset", subset, "n files", len(files)
-        files = map(lambda x: x.split(x.replace('.mhd',''),'/')[-1], files)
+        files = map(lambda x: x.replace('.mhd','').split('/')[-1], files)
         subset_dict[subset] = files
 
     joblib.dump(subset_dict, '../config/subset_to_filenames.pkl')