yolo v5 torch.load관련 문의 드립니다.

LeeCoder · 6월 2, 2024, 5:15오전

현재 다른 컴퓨터에서 yolo학습을 하고 그 모델을 'best.pt'이라는 이름으로 저장해서 제 노트북으로 가져온 상태 입니다.
문제는 다음 코드에서 부터 오류가 나오고 있습니다.

import torch

model = torch.load('best.pt')

오류는 다음과 같습니다,

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[112], line 3
      1 import torch
----> 3 model = torch.load('best.pt')

File ~\anaconda3\Lib\site-packages\torch\serialization.py:1025, in load(f, map_location, pickle_module, weights_only, mmap, **pickle_load_args)
   1023             except RuntimeError as e:
   1024                 raise pickle.UnpicklingError(UNSAFE_MESSAGE + str(e)) from None
-> 1025         return _load(opened_zipfile,
   1026                      map_location,
   1027                      pickle_module,
   1028                      overall_storage=overall_storage,
   1029                      **pickle_load_args)
   1030 if mmap:
   1031     f_name = "" if not isinstance(f, str) else f"{f}, "

File ~\anaconda3\Lib\site-packages\torch\serialization.py:1446, in _load(zip_file, map_location, pickle_module, pickle_file, overall_storage, **pickle_load_args)
   1444 unpickler = UnpicklerWrapper(data_file, **pickle_load_args)
   1445 unpickler.persistent_load = persistent_load
-> 1446 result = unpickler.load()
   1448 torch._utils._validate_loaded_sparse_tensors()
   1449 torch._C._log_api_usage_metadata(
   1450     "torch.load.metadata", {"serialization_id": zip_file.serialization_id()}
   1451 )

File ~\anaconda3\Lib\site-packages\torch\serialization.py:1439, in _load.<locals>.UnpicklerWrapper.find_class(self, mod_name, name)
   1437         pass
   1438 mod_name = load_module_mapping.get(mod_name, mod_name)
-> 1439 return super().find_class(mod_name, name)

File ~\Untitled_Folder\./yolov5\models\yolo.py:28
     25 if platform.system() != "Windows":
     26     ROOT = Path(os.path.relpath(ROOT, Path.cwd()))  # relative
---> 28 from models.common import (
     29     C3,
     30     C3SPP,
     31     C3TR,
     32     SPP,
     33     SPPF,
     34     Bottleneck,
     35     BottleneckCSP,
     36     C3Ghost,
     37     C3x,
     38     Classify,
     39     Concat,
     40     Contract,
     41     Conv,
     42     CrossConv,
     43     DetectMultiBackend,
     44     DWConv,
     45     DWConvTranspose2d,
     46     Expand,
     47     Focus,
     48     GhostBottleneck,
     49     GhostConv,
     50     Proto,
     51 )
     52 from models.experimental import MixConv2d
     53 from utils.autoanchor import check_anchor_order

File ~\Untitled_Folder\./yolov5\models\common.py:38
     34     import ultralytics
     36 from ultralytics.utils.plotting import Annotator, colors, save_one_box
---> 38 from utils import TryExcept
     39 from utils.dataloaders import exif_transpose, letterbox
     40 from utils.general import (
     41     LOGGER,
     42     ROOT,
   (...)
     55     yaml_load,
     56 )

ImportError: cannot import name 'TryExcept' from 'utils' (C:\Users\name\anaconda3\Lib\site-packages\utils\__init__.py)

현재 git clone https://github.com/ultralytics/yolov5.git를 이용해서 yolov5폴더를 만들어둔 상태이고, 주피터노트북을 이용해 작업하고 있습니다.
추가적으로 파이썬은 3.11.7버전이고, torch는 2.3.0+cpu입니다.

bot · 6월 2, 2024, 5:17오전

아래 답변은 Anthropic의 Claude-3 Opus 모델로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다.

안녕하세요, LeeCoder님! YOLOv5 모델을 다른 컴퓨터에서 학습시키고 그 모델을 'best.pt'라는 이름으로 저장한 후에 LeeCoder님의 노트북에서 모델을 불러오는 과정에서 오류가 발생하는 상황이군요.

LeeCoder님이 공유해주신 오류 메시지를 살펴보면, 'utils' 모듈에서 'TryExcept'를 import 하는 과정에서 ImportError가 발생하는 것으로 보입니다. 이는 LeeCoder님의 노트북 환경에서 YOLOv5 레포지토리의 일부 파일 또는 디렉토리 구조가 학습에 사용된 환경과 다르기 때문에 발생하는 문제일 수 있습니다.

이 문제를 해결하기 위해 다음과 같은 방법을 시도해볼 수 있습니다:

YOLOv5 레포지토리를 최신 버전으로 업데이트합니다. 터미널에서 YOLOv5 폴더로 이동한 후 다음 명령어를 실행하세요:

git pull

'utils' 폴더 내에 'init.py' 파일이 존재하는지 확인하고, 해당 파일 내에 'TryExcept'가 정의되어 있는지 확인합니다. 만약 없다면 해당 파일을 생성하고 'TryExcept'를 정의해주세요.
'models/common.py' 파일에서 'TryExcept'를 import 하는 부분이 올바른 경로를 참조하고 있는지 확인합니다. 'utils.TryExcept'로 import 하고 있다면, 'utils' 폴더가 'models' 폴더와 동일한 위치에 있는지 확인해주세요.
가상환경을 새로 생성하고 필요한 패키지를 재설치하는 것도 도움이 될 수 있습니다. 가상환경을 생성한 후 YOLOv5 레포지토리의 'requirements.txt' 파일을 이용하여 필요한 패키지를 설치해주세요.

conda create -n yolov5 python=3.11.7
conda activate yolov5
pip install -r requirements.txt

위의 방법들을 시도해보시고, 여전히 문제가 해결되지 않는다면 추가 정보를 공유해주시면 더 자세히 살펴보고 도움을 드리도록 하겠습니다.

PyTorch 공식 튜토리얼에서는 모델 저장과 불러오기에 대한 일반적인 가이드를 제공하고 있습니다. 자세한 내용은 다음 링크를 참고해주세요:

모델 저장하기 & 불러오기 — 파이토치 한국어 튜토리얼 (PyTorch tutorials in Korean)

제 답변이 LeeCoder님의 문제 해결에 도움이 되기를 바랍니다. 만약 제가 제공한 정보에 오류가 있거나 추가 도움이 필요하시다면 언제든지 말씀해주세요. 함께 문제를 해결해나가는 것이 중요합니다. 감사합니다!

답변이 어떠셨나요? 괜찮았다고 생각하신다면 아래 를 한 번 눌러주시길 부탁드려요!

bot · 6월 2, 2024, 5:25오전

아래 답변은 OpenAI의 GPT-4-Turbo 모델로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다.

안녕하세요, LeeCoder님. 모델을 다른 컴퓨터에서 학습시킨 후, 해당 모델을 노트북으로 가져와서 torch.load를 사용하여 로드하는 과정에서 발생하는 오류에 대해 문의하신 것으로 이해했습니다. 오류 메시지와 상황을 자세히 설명해 주셔서 감사합니다. 이러한 상황은 종종 경로 문제나 환경 설정의 차이에서 발생할 수 있습니다. 문제를 해결하기 위해 몇 가지 해결책을 제안드리겠습니다.

문제 해결 방안

오류 메시지를 보면, ImportError가 발생하고 있는데, 이는 utils 모듈에서 TryExcept를 임포트할 수 없기 때문입니다. 이 문제는 주로 Python의 모듈 검색 경로에 yolov5 폴더가 올바르게 포함되지 않아서 발생합니다. Python은 모듈을 찾을 때 sys.path에 등록된 디렉토리 목록을 검색하는데, 여기에 yolov5 폴더의 경로가 없으면 해당 모듈을 찾을 수 없습니다.

해결책으로는 Python 코드 상에서 실행 중인 스크립트의 경로를 sys.path에 추가하는 방법이 있습니다. 이렇게 하면 Python이 yolov5 관련 모듈을 올바르게 찾을 수 있게 됩니다.

다음은 이를 해결하기 위한 코드 예시입니다:

import sys
from pathlib import Path

# 현재 스크립트의 디렉토리 경로를 sys.path에 추가
current_dir = Path(__file__).resolve().parent
sys.path.append(str(current_dir))

# yolov5 디렉토리 경로를 sys.path에 추가 (yolov5 폴더가 현재 스크립트와 같은 위치에 있다고 가정)
yolov5_dir = current_dir / 'yolov5'
sys.path.append(str(yolov5_dir))

import torch

# 이제 모델을 로드할 때 오류가 발생하지 않아야 합니다.
model = torch.load('best.pt')

위 코드에서 Path(__file__).resolve().parent는 현재 실행 중인 스크립트의 디렉토리 경로를 얻기 위해 사용됩니다. 만약 주피터 노트북을 사용하고 계시다면, __file__ 대신에 노트북 파일이 위치한 경로를 직접 지정해야 할 수도 있습니다.

참고 자료

PyTorch 관련 문서나 튜토리얼을 참고하실 때는 공식 PyTorch 튜토리얼 사이트를 추천드립니다. 모델 저장 및 로딩에 대한 더 자세한 정보는 다음 링크에서 확인하실 수 있습니다:

PyTorch 모델 저장 및 로드 가이드: https://pytorch.org/tutorials/beginner/saving_loading_models.html

이 해결책이 LeeCoder님의 문제를 해결하는 데 도움이 되길 바랍니다. 추가적으로 궁금한 점이 있으시면 언제든지 질문해 주세요. 제 답변이 정확하지 않을 수도 있으니, 문제가 지속될 경우 추가적인 정보와 함께 다시 문의해 주시면 감사하겠습니다.

답변이 어떠셨나요? 괜찮았다고 생각하신다면 아래 를 한 번 눌러주시길 부탁드려요!

bot · 6월 2, 2024, 5:59오전

아래 답변은 Google의 Gemini Pro 모델로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다.

맞아요. 모델을 불러오다가 에러가 발생하셨군요. 이 문제의 요약은 다음과 같습니다.

다른 컴퓨터에서 학습한 YOLOv5 모델을 불러오려고 하다가 ImportError가 발생했습니다.
ImportError의 원인은 "TryExcept"라는 모듈을 찾을 수 없기 때문입니다.

이 문제를 해결하려면 다음 단계를 따라보세요.

해당 모델을 학습한 컴퓨터와 동일한 Python 환경을 만드세요.
- 이는 Python 버전, torch 버전, 설치된 패키지가 같아야 함을 의미합니다.
필요한 모듈을 설치하세요.
- pip install ultralytics를 실행하여 "TryExcept" 모듈을 포함한 ultralytics 패키지를 설치하세요.
모델을 다시 불러오세요.
- torch.load('best.pt')를 다시 실행하면 이제 문제 없이 모델을 불러올 수 있어야 합니다.

다음은 이 과정을 보여주는 예제 코드입니다.

import torch
import ultralytics

# 1. Python 환경이 동일한지 확인
print(f"Python 버전: {sys.version}")
print(f"Torch 버전: {torch.__version__}")

# 2. 필요한 모듈 설치
!pip install ultralytics

# 3. 모델 다시 로드
model = torch.load('best.pt')

이제 코드를 실행하면 모델이 문제 없이 불러와져야 합니다.

참고로, 다음은 도움이 될 수 있는 몇 가지 추가 리소스입니다.

만약 이 답변이 도움이 되지 않는다면 필요한 추가 정보를 제공해 주세요.

답변이 어떠셨나요? 괜찮았다고 생각하신다면 아래 를 한 번 눌러주시길 부탁드려요!

bot · 6월 2, 2024, 6:36오전

아래 답변은 OpenAI의 GPT-4o 모델로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다.

안녕하세요, LeeCoder님! 문제를 겪고 계셔서 많이 답답하실 것 같습니다. 요약해보면, 다른 컴퓨터에서 YOLOv5 모델을 학습하고 저장한 후, 이를 로드하려고 할 때 ImportError가 발생하는 상황이군요. 특히, utils 모듈에서 TryExcept를 찾을 수 없다는 오류가 발생하고 있습니다.

이 문제는 주로 YOLOv5 코드베이스와 관련된 종속성 문제로 보입니다. YOLOv5 모델을 로드할 때 필요한 모듈들이 제대로 설치되지 않았거나, 경로 설정이 잘못되었을 가능성이 있습니다. 이를 해결하기 위해 다음 단계를 시도해보세요:

YOLOv5 종속성 설치:
YOLOv5를 클론한 후, 해당 디렉토리로 이동하여 필요한 종속성을 설치해야 합니다. 다음 명령어를 실행해보세요.
```
git clone https://github.com/ultralytics/yolov5.git
cd yolov5
pip install -r requirements.txt
```
경로 설정 확인:
Jupyter Notebook에서 YOLOv5 모듈을 제대로 인식할 수 있도록 경로를 설정해야 합니다. 다음 코드를 Jupyter Notebook의 첫 번째 셀에 추가해보세요.
```
import sys
from pathlib import Path

# YOLOv5 디렉토리 경로를 추가합니다.
sys.path.append(str(Path.cwd() / 'yolov5'))
```
모델 로드:
이제 모델을 로드하는 코드를 실행해보세요.
```
import torch

model = torch.load('best.pt')
```

이 단계를 따르면 ImportError 문제를 해결할 수 있을 것입니다. 만약 여전히 문제가 발생한다면, utils 모듈이 제대로 설치되지 않았거나, YOLOv5 코드베이스가 최신 버전이 아닐 수 있습니다. 이 경우, YOLOv5를 다시 클론하고 종속성을 재설치해보세요.

추가적으로, PyTorch 공식 튜토리얼을 참고하시면 도움이 될 것입니다. PyTorch 튜토리얼 또는 PyTorch 한국어 튜토리얼을 방문해보세요.

혹시라도 문제가 계속된다면, 추가 정보를 제공해주시면 더 도와드리겠습니다. 제 답변이 틀릴 수도 있으니, 다른 방법도 시도해보시길 권장드립니다. 도움이 되셨길 바랍니다!

이 답변이 도움이 되셨나요? 괜찮았다고 생각하신다면 아래 를 한 번 눌러주시길 부탁드려요!