다운로드 받은 yolov5s.pt 파일을 로드하는 방법이 어떤건가요?

묻고 답하기
1

code

import torch
from PIL import Image
import time
#from models.experimental import attempt_load

weights='../alg/yolov5s.pt'

# Model
#model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
#model = attempt_load(weights)
model = torch.load(weights)

# Image
img = '../../data/test_car.png'

# Resize
img = Image.open(img)

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

# Inference
start_time_model = time.time()
#results = loadded_model(img_resize)
results = model(img_resize)
#results = model(img)
model_elapsed_time = time.time()-start_time_model
print("model time = "+ str(model_elapsed_time))

results.show()

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

error 내용
Traceback (most recent call last):
File “pytorch_yolov5.py”, line 10, in
model = torch.load(weights)
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/serialization.py”, line 592, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “/home/nvidia/.local/lib/python3.6/site-packages/torch/serialization.py”, line 851, in _load
result = unpickler.load()
ModuleNotFoundError: No module named ‘models’

입니다.

code에서 …/alg/ 여기에 .pt file이 다 모여있는데 load 방법을 잘 모르겠습니다.
구글에서 검색을 하고 여기 포럼에서도 검색을 한 뒤에 적용을 해봐도 계속 에러가 납니다.

어디가 문제인지 혹시 알려주실 수 있을까요…?

2

안녕하세요. 혹시 .pt 파일을 어디서 다운 받으셨는지 알 수 있을까요?
단순한 경로문제 일수도 있을 것 같다는 생각이 듭니다.