crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add input data check

Browse Source
This commit is contained in:
darkliang
2022-01-18 19:52:38 +08:00
parent 19ed960b0a
commit 7c33b33440
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/data/dataset.py
+8 -6
View File
@@ -41,13 +41,14 @@ class DataSet(tordata.Dataset):
f.close() f.close()
else: else:
raise ValueError('- Loader - just support .pkl !!!') raise ValueError('- Loader - just support .pkl !!!')
# if len(_) >= 200:
# _ = _[:200]
data_list.append(_) data_list.append(_)
for data in data_list: for idx, data in enumerate(data_list):
if len(data) != len(data_list[0]): if len(data) != len(data_list[0]):
raise ValueError('Each input data should have the same length.') raise ValueError(
'Each input data({}) should have the same length.'.format(paths[idx]))
if len(data) == 0:
raise ValueError(
'Each input data({}) should have at least one element.'.format(paths[idx]))
return data_list return data_list
def __getitem__(self, idx): def __getitem__(self, idx):
@@ -116,7 +117,8 @@ class DataSet(tordata.Dataset):
seq_dirs, data_in_use) if use_bl] seq_dirs, data_in_use) if use_bl]
seqs_info_list.append([*seq_info, seq_dirs]) seqs_info_list.append([*seq_info, seq_dirs])
else: else:
msg_mgr.log_debug('Find no .pkl file in %s-%s-%s.'%(lab, typ, vie)) msg_mgr.log_debug(
'Find no .pkl file in %s-%s-%s.' % (lab, typ, vie))
return seqs_info_list return seqs_info_list
self.seqs_info = get_seqs_info_list( self.seqs_info = get_seqs_info_list(