Commit 79534f0a authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Fix the FrameIndexes keys in dataloader

parent 064ad286
Pipeline #27876 failed with stage
in 7 minutes and 36 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 len(f.split('_'))==2]
# elements of tuples in the below list are as follows:
# a filename a key is extracted from,
# a label corresponding to the file,
Markdown is supported
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