AlibabaResearch / AlibabaResearch/efficientteacher

pt文件转换问题

Offen
#43 9 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
813
Forks
126
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

我在训练好efficient_yolol.pt之后,使用convert_efficient_to_yolov5转换成标准的yolov5l.pt,但在使用yolov5官方的detect.py时,出现了以下bug:
Traceback (most recent call last):
File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 307, in
main(opt)
File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 302, in main
run(* * vars(opt))
File "/usr/local/anaconda3/envs/***/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, * * kwargs)
File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 82, in run
model = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location=device)
File "/home/zzz/Desktop/workspace/yolov5/models/experimental.py", line 94, in attempt_load
ckpt = torch.load(attempt_download(w), map_location=map_location) # load
File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 607, in load
return _load(opened_zipfile, map_location, pickle_module, * * pickle_load_args)
File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load
result = unpickler.load()
File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 875, in find_class
return super().find_class(mod_name, name)
**AttributeError: Can't get attribute 'DetectionModel' on **

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.