yolov5 웨이트파일 로드하기

[shj]

import torch
import time

weights=‘best.pt’

model = torch.load(weights)

img = ‘C:/Users/1/Desktop/yolov5-master/pythonProject3/venv/lib/image.jpg’

img_resize = img.resize((640, 480))

start_time_model = time.time()

results = model(img_resize)

model_elapsed_time = time.time()-start_time_model
print("model time = "+ str(model_elapsed_time))

results.show()

print(results.pandas().xyxy[0])

파이토치 허브에서 수정한 코드인데요 다음과 같이 파일을 실행하면

Traceback (most recent call last):
File “C:/Users/1/Desktop/yolov5-master/pythonProject3/venv/Lib/deep.py”, line 6, in
model = torch.load(weights)
File “C:\Users\1\Desktop\yolov5-master\pythonProject3\venv\lib\site-packages\torch\serialization.py”, line 789, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “C:\Users\1\Desktop\yolov5-master\pythonProject3\venv\lib\site-packages\torch\serialization.py”, line 1131, in _load
result = unpickler.load()
File “C:\Users\1\Desktop\yolov5-master\pythonProject3\venv\lib\site-packages\torch\serialization.py”, line 1124, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named ‘models’

이렇게 오류가 뜨는데 models 모듈이 없다는게 무슨뜻인지 모르겠습니다

[cpprhtn (Junwon Lee)]

아래 issues를 통해 해결할 수 있을듯합니다.

ModuleNotFoundError: No module named 'models'