Commit 7007d842 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Merge branch 'fix_keys' into 'master'

Fix the FrameIndexes keys in dataloader

See merge request !23
parents 064ad286 51ea4133
Pipeline #27886 failed with stages
in 2 minutes and 10 seconds
......@@ -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,
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment