saundersp/ViolaJones
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python : added comment for potential indices in unit test

Browse Source
This commit is contained in:
saundersp 2024-04-28 00:26:31 +02:00
parent 434ce20374
commit f7ac38b93a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/common.py
View File

@ -60,11 +60,13 @@ def unit_test(TS: List[int], labels: List[str] = ['CPU', 'GPU', 'PY', 'PGPU'], t
title = f'X_{set_name}_feat_argsort'
print(f'Loading {title}...', file = stderr, end = '\r')
feat = None
#indices = pickle_multi_loader(['indices'], OUT_DIR)[0]
bs = []
for label in labels:
if feat is None:
feat_tmp = pickle_multi_loader([f'X_{set_name}_feat_{label}'], OUT_DIR)[0]
if feat_tmp is not None:
#feat = feat_tmp[indices]
feat = feat_tmp
bs.append(pickle_multi_loader([f'{title}_{label}'], OUT_DIR)[0])