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 모듈이 없다는게 무슨뜻인지 모르겠습니다
