jinql
/
Yolov5-deepsort-inference
mirror of https://github.com/Sharpiless/Yolov5-deepsort-inference
1
0
Fork 0
Code Issues Releases Wiki Activity

增加对yolov5 4.0的支持

pull/28/head
LYP 2021-02-02 17:06:25 +08:00
parent 62a205f741
commit 1475a96393
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
models/experimental.py
View File

@ -134,7 +134,10 @@ def attempt_load(weights, map_location=None):
model = Ensemble() model = Ensemble()
for w in weights if isinstance(weights, list) else [weights]: for w in weights if isinstance(weights, list) else [weights]:
attempt_download(w) attempt_download(w)
model.append(torch.load(w, map_location=map_location)['model'].float().fuse().eval()) # load FP32 model try:
model.append(torch.load(w, map_location=map_location)['model'].float().fuse().eval()) # load FP32 model
except Exception:
model.append(torch.load(w, map_location=map_location).float().fuse().eval()) # load FP32 model
# Compatibility updates # Compatibility updates
for m in model.modules(): for m in model.modules():