python 中的 mtcnn 中未定义名称“MTCNN”

Question

我安装了mtcnn包，可以看到：

# confirm mtcnn was installed correctly
import mtcnn
# print version
print(mtcnn.__version__)

结果：

0.0.9

然后我使用以下内容：

# prepare model
model = MTCNN()
# detect face in the image
faces = model.detect_faces(pixels)
# extract details of the face
x1, y1, width, height = faces[0]['box']

结果：

1 # prepare model
----> 2 model = MTCNN()
      3 # detect face in the image
      4 faces = model.detect_faces(pixels)
      5 # extract details of the face

NameError: name 'MTCNN' is not defined

Answer 1

该文档是一个很好的起点： https: //github.com/ipazc/mtcnn

from mtcnn.mtcnn import MTCNN
import cv2


model = MTCNN()

Answer 2

有时，人们所做的是将他们正在处理的文件命名为 mtcnn.py ，这会导致此错误。将文件 mtcnn.py 的名称更改为其他名称即可。