Skip to content
Snippets Groups Projects

Fix the FrameIndexes keys in dataloader

Merged Anjith GEORGE requested to merge fix_keys into master
1 file
+ 3
0
Compare changes
  • Side-by-side
  • Inline
@@ -76,6 +76,9 @@ 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 f!='FrameIndexes' ]
# elements of tuples in the below list are as follows:
# a filename a key is extracted from,
# a label corresponding to the file,
Loading