Remove path from cand names

gzuidhof · Jun 20, 2016 · 3a7cc0d · 3a7cc0d
1 parent 44ed601
commit 3a7cc0d
Showing 1 changed file with 1 addition and 1 deletion.
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.replace('.mhd',''), files)
+        files = map(lambda x: x.split(x.replace('.mhd',''),'/')[-1], files)
         subset_dict[subset] = files
 
     joblib.dump(subset_dict, '../config/subset_to_filenames.pkl')