首页 > 解决方案 > 在 Pytorch 教程期间发生 ModuleNotFoundError: 'pycocotools._mask'

问题描述

您好,我是 Pytorch 的新手,我一直在尝试完成本教程。[ https://github.com/pytorch/tutorials/blob/master/intermediate_source/torchvision_tutorial.rst]

我正在使用 Visual Studio Code,conda 安装了 Pytorch。最初的问题是它缺少模块引擎。但是这个问题在这里解决了:那么我们如何获得references/detection/文件夹呢?我们应该下载和安装什么?我已经在我的环境中安装了 pytorch、torchvision,但我找不到这些文件。谢谢我误读了一步。所以我下载了 pycocotools 并将其复制到项目目录中,并将 vision/detection/ .py 文件也复制到其中。这样做它解决了这个问题,但它现在给出的错误是。

(base) C:\Users\Sean\Desktop\Project\Test\Tutorial>D:/Anaconda/python.exe c:/Users/Sean/Desktop/Project/Test/Tutorial/tv-training-code.py
Traceback (most recent call last):
  File "c:/Users/Sean/Desktop/Project/Test/Tutorial/tv-training-code.py", line 13, in <module>
    from engine import train_one_epoch, evaluate
  File "c:\Users\Sean\Desktop\Project\Test\Tutorial\engine.py", line 8, in <module>
    from coco_utils import get_coco_api_from_dataset
  File "c:\Users\Sean\Desktop\Project\Test\Tutorial\coco_utils.py", line 9, in <module>
    from pycocotools import mask as coco_mask
  File "c:\Users\Sean\Desktop\Project\Test\Tutorial\pycocotools\mask.py", line 3, in <module>
    import pycocotools._mask as _mask
ModuleNotFoundError: No module named 'pycocotools._mask'

我不太确定这里的问题是什么或如何解决它。鉴于存在 _mask.pyx 并且是我认为正在导入的内容。正如我所说,我不确定问题是什么,但我将不胜感激。

Python 版本:3.7.4 Pytorch:1.2.0(Cuda 10)

标签: pythonpython-3.xvisual-studio-codepytorchtorchvision

解决方案


问题是您复制了文件pycocotools而不是安装它们。以结尾的.pyx文件是需要安装到扩展模块中的 Cython 文件(在 Windows 上,这些文件是一个.pyd文件)。如果您安装软件包而不是应该解决问题的文件副本。


推荐阅读