Skip to content
Snippets Groups Projects

Fix the FrameIndexes keys in dataloader

Merged Anjith GEORGE requested to merge fix_keys into master
2 unresolved threads
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -77,7 +77,7 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type
file_keys = list(f_h5py.keys())
#removes the 'FrameIndexes' key
file_keys=[f for f in file_keys if len(f.split('_'))==2]
file_keys=[f for f in file_keys if f!='FrameIndexes' ]
# elements of tuples in the below list are as follows:
# a filename a key is extracted from,
Loading