mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

FeatureExtractor: avoid loading weights twice

Browse Source 
When passing a model_path to the FeatureExtractor, weights are loaded twice.
One the pretrained once and then the given ones. Avoid the unnecessary load.
This commit is contained in:
Helmut Grohne 2022-01-19 09:25:20 +01:00 committed by Helmut Grohne
parent 6987d99074
commit 42185c4beb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/utils/feature_extractor.py
View File

@ -71,7 +71,7 @@ class FeatureExtractor(object):
model = build_model(
model_name,
num_classes=1,
pretrained=True,
pretrained=not (model_path and check_isfile(model_path)),
use_gpu=device.startswith('cuda')
)
model.eval()