인프런 커뮤니티 질문&답변

안녕하세요? 선생님

 

개발한 YOLOv5 모델을 하드에 저장한 후 불러와서 사용하려고 합니다.

 

튜토리얼에 따라 다음과 같이 하면 되지만, 이 경우 인터넷 접속이 되어 있어야 합니다.

model = torch.hub.load('ultralytics/yolov5', 'custom', path='mybest.pt')

 

그래서 GitHub 사용하지 않게 다음과 같이 수정했습니다.

model=torch.hub.load('yolov5', 'custom', path='mybest.pt', source='local')

 

그런데, 이렇게 하면 속도가 1/3로 줄어드는 현상이 나타났습니다.

인터넷 없이, 속도는 유지하면서 추론하는 방법은 없을까요?