YOLO报错No module named ‘ultralytics‘

Dneccc

已于 2024-11-29 08:32:14 修改

阅读量8.2k

点赞数 8

CC 4.0 BY-SA版权

文章标签：深度学习人工智能

于 2024-11-29 08:12:44 首次发布

本文链接：https://blog.csdn.net/Dneccc/article/details/144125996

当你自己写脚本

from ultralytics import YOLO

脚本没有放在ultralytics文件夹同级目录下时：

|- yolo

|- train.py

|- ultralytics

运行train.py可能会出现报错ModuleNotFoundError: No module named 'ultralytics'

解决办法：

import sys
import os
current_dir = os.getcwd()  # 获取当前工作目录
current_file_path = os.path.abspath(__file__)  # 获取当前文件的绝对路径
current_fadir1 = os.path.dirname(current_file_path)  # 获取当前文件的1级父级目录绝对路径
current_fadir2 =  os.path.dirname(current_fadir1)  # 获取当前文件的2级父级目录绝对路径
sys.path.extend([current_dir, current_file_path, current_fadir1, current_fadir2])  # 将当前工作目录添加到 sys.path 中

from ultralytics import YOLO