mirrors/PyRetri
1
0
Fork 0
mirror of https://github.com/PyRetri/PyRetri.git synced 2025-06-03 14:49:50 +08:00
Code Issues Projects Releases Wiki Activity

convert to cpu()

Browse Source
This commit is contained in:
hubenyi 2020-06-29 17:03:21 +08:00
parent b2e189d483
commit 53bd3b3768
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main/single_index.py
View File

@ -50,7 +50,7 @@ def main():
stacked_feature = list() stacked_feature = list()
for name in cfg.index.feature_names: for name in cfg.index.feature_names:
assert name in img_fea_info[0], "invalid feature name: {} not in {}!".format(name, img_fea_info[0].keys()) assert name in img_fea_info[0], "invalid feature name: {} not in {}!".format(name, img_fea_info[0].keys())
stacked_feature.append(img_fea_info[0][name]) stacked_feature.append(img_fea_info[0][name].cpu())
img_fea = np.concatenate(stacked_feature, axis=1) img_fea = np.concatenate(stacked_feature, axis=1)
# load gallery features # load gallery features