Skip to content
Snippets Groups Projects
Commit 51ea4133 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Quick fix, for keys

parent 79534f0a
No related branches found
No related tags found
1 merge request!23Fix the FrameIndexes keys in dataloader
Pipeline #27878 failed
Stage: build
Hide whitespace changes
Inline Side-by-side
bob/learn/pytorch/datasets/data_folder.py
+ 1
− 1
View file @ 51ea4133
...@@ -77,7 +77,7 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type ...@@ -77,7 +77,7 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type
file_keys = list(f_h5py.keys()) file_keys = list(f_h5py.keys())
#removes the 'FrameIndexes' key #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: # elements of tuples in the below list are as follows:
# a filename a key is extracted from, # a filename a key is extracted from,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment