### 解决YOLO自定义模块未找到问题 当在Ubuntu服务器上使用Anaconda部署Python项目并遇到`ModuleNotFoundError`错误时，通常是因为项目的内部包或文件未能被正确加载到Python运行环境中。对于YOLO项目中的自定义模块找不到的问题，可以通过以下方法解决： #### 方法一：调整PYTHONPATH环境变量 通过设置`PYTHONPATH`来包含YOLO项目的根目录，从而让Python能够识别项目内的自定义模块。 ```bash export PYTHONPATH=$PYTHONPATH:/path/to/your/yolo_project/ ``` 这种方法虽然简单，但在多项目环境下可能导致冲突，因此不作为首选解决方案[^1]。 --- #### 方法二：动态修改`sys.path` 可以在脚本开头手动将YOLO项目的路径添加到`sys.path`中，以便Python能够在运行时定位到所需的模块。 ```python import sys sys.path.append('/path/to/your/yolo_project/') ``` 尽管此方法有效，但它需要在每个涉及该问题的脚本中重复执行，增加了维护成本，故也不推荐长期使用。 --- #### 方法三：利用`.pth`文件扩展`site-packages` 最推荐的方法是在当前虚拟环境的`site-packages`目录下创建一个`.pth`文件，并将其指向YOLO项目的根目录。具体操作如下： 1. **切换至虚拟环境的`site-packages`目录** ```bash cd $(dirname $(which python))/../lib/python$(python --version | cut -d ' ' -f 2)/site-packages/ ``` 2. **创建一个新的`.pth`文件** 使用任意名称（如`yolo.pth`），并将YOLO项目的绝对路径写入其中。 ```bash echo /path/to/your/yolo_project/ > yolo.pth ``` 3. **验证配置是否生效** 运行以下命令确认新路径已被成功加载： ```python import sys print(sys.path) ``` 上述步骤完成后，Python解释器会在解析模块时自动扫描指定路径下的所有子模块，从而避免`ModuleNotFoundError`的发生。 --- #### 示例代码片段 假设您的YOLO项目结构如下： ``` /path/to/your/yolo_project/ ├── custom_module.py └── yolov10m.pt ``` 如果尝试调用`custom_module`时报错，则按照前述方法处理后即可正常运行以下代码： ```python from custom_module import some_function model = YOLO("yolov10m.pt") results = model.predict("image.jpg", save=False, save_txt=True, verbose=False, conf=0.75) boxes = results[0].boxes confidence, class_ids = boxes.conf, boxes.cls.int() rects = boxes.xyxy.int() for ind in range(boxes.shape[0]): print(some_function(class_ids[ind].item()), confidence[ind].item(), rects[ind].tolist()) ``` --- ### 总结 针对YOLO自定义模块未找到的问题，建议优先采用第三种方法——即通过`.pth`文件的方式扩展`site-packages`路径。这种方式不仅高效且易于管理，还不会影响其他项目的独立性。 ---