python-3.x - 如何修复 "AttributeError: module ' pycocotools' has no attribute 'mask' "python 中的错误

Question

即使我正确安装了 pycocotools，我仍面临一些问题。此外，在已经安装了 pycocotools 的 colab 上进行了尝试。

安装方法:

!git clone https://github.com/cocodataset/cocoapi.git
%cd /content/cocoapi/PythonAPI
!make
!sudo make install
!sudo python setup.py install
%cd /content

AttributeError                            Traceback (most recent call last)

<ipython-input-11-6a1416ecd890> in <module>()
----> 1 new_mask = get_mask(label['segmentations'], mask)

/content/datasets/coco.py in get_mask(segmentations, mask)
     17 def get_mask(segmentations, mask):
     18     for segmentation in segmentations:
---> 19         rle = pycocotools.mask.frPyObjects(segmentation, mask.shape[0], mask.shape[1])
     20         mask[pycocotools.mask.decode(rle) > 0.5] = 0
     21     return mask

AttributeError: module 'pycocotools' has no attribute 'mask'

抛出此错误的行:pycocotools.mask.frPyObjects(分割, mask.shape[0], mask.shape[1])

Answer 1

如果你只是导入了pycocotools，mask模块不会被导入。

尝试显式导入掩码:

import pycocotools.mask