Traceback (most recent call last): File "/root/miniconda3/envs/test/bin/yolo", line 8, in <module> sys.exit(entrypoint()) File "/root/miniconda3/envs/test/lib/python3.8/site-packages/ultralytics/yolo/cfg/init.py", line 370, in entrypoint model = YOLO(model, task=task) File "/root/miniconda3/envs/test/lib/python3.8/site-packages/ultralytics/yolo/engine/model.py", line 105, in init self._new(model, task) File "/root/miniconda3/envs/test/lib/python3.8/site-packages/ultralytics/yolo/engine/model.py", line 138, in _new self.model = TASK_MAP[self.task][0](cfg_dict, verbose=verbose and RANK == -1) # build model File "/root/miniconda3/envs/test/lib/python3.8/site-packages/ultralytics/nn/tasks.py", line 189, in init self.model, self.save = parse_model(deepcopy(self.yaml), ch=ch, verbose=verbose) # model, savelist File "/root/miniconda3/envs/test/lib/python3.8/site-packages/ultralytics/nn/tasks.py", line 491, in parse_model m = getattr(torch.nn, m[3:]) if 'nn.' in m else globals()[m] # get module KeyError: 'space_to_depth' Sentry is attempting to send 2 pending events Waiting up to 2 seconds Press Ctrl-C to quit
时间: 2023-06-30 13:17:38 浏览: 520
linux 下 pip3 报错“ File “/usr/bin/pip3”, line 9, in from pip import mainImportError: canno”处理
5星 · 资源好评率100%
根据错误信息,看起来问题出在 `ultralytics` 库中的 `tasks.py` 文件中,具体是 `getattr(torch.nn, m[3:]) if 'nn.' in m else globals()[m]` 这一行代码中。这里尝试从 `torch.nn` 或者当前的全局命名空间中获取一个名为 `space_to_depth` 的模块,但是没有找到,导致了 `KeyError` 异常。
这个问题可能是因为 `space_to_depth` 没有被正确地安装,或者没有被正确地导入。您可以尝试检查一下 `space_to_depth` 是否已经被正确安装,并且已经被正确地导入到当前的环境中。如果您已经正确安装了 `space_to_depth`,那么您可以尝试手动导入该模块,以确保它被正确地加载了。例如,您可以在 Python 交互环境中执行以下代码:
```python
import space_to_depth
```
如果这里出现了任何错误提示,那么说明 `space_to_depth` 没有被正确地加载。您可以尝试重新安装该模块,或者检查一下您的 Python 环境是否正确配置。
阅读全文