在Pytorch教程中,发生ModuleNotFoundError:“ pycocotools._mask”

时间:2019-12-04 14:43:58

标签: python python-3.x visual-studio-code pytorch torchvision

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

我正在使用Visual Studio Code,conda安装了Pytorch。最初的问题是缺少模块引擎。但是这个问题在这里解决了: 那么,我们如何获得参考/检测/文件夹?我们应该下载并安装什么?我已经在我的环境中安装了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 火炬:1.2.0(Cuda 10)

1 个答案:

答案 0 :(得分:0)

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