'NoneType' object has no attribute 'size' - 如何使用mtcnn进行人脸检测？

Question

我正在尝试使用 pytorch 在神经网络（facenet 网络）中实时构建人脸识别，并使用 MTCNN 进行人脸检测我试过这个来实时检测人脸（从网络摄像头）但是没有用读取帧然后通过 mtcnn 检测器

import cv2  
capture = cv2.VideoCapture(0)  
while(True):      
   ret, frame = capture.read()     
   frames_tracked = []      
   print('\rTracking frame: {}'.format(i + 1), end='')     
   boxes,_ = mtcnn.detect(frame)     
   frame_draw = frame.copy()     
   draw = ImageDraw.Draw(frame_draw)     
   for box in boxes:
       draw.rectangle(box.tolist(), outline=(255, 0, 0), width=6)          
       frames_tracked.append(frame_draw.resize((640, 360), Image.BILINEAR))      
   d = display.display(frames_tracked[0], display_id=True)     
   i = 1     
   try:        
      while True:
          d.update(frames_tracked[i % len(frames_tracked)]) 
          i += 1     
   except KeyboardInterrupt:
      pass  
   if cv2.waitKey('q') == 27:     
     break  

  capture.release() 
  cv2.destroyAllWindows()

但是会出现这个错误:

这是整个追溯http://dpaste.com/0HR58RQ

AttributeError: 'NoneType' object has no attribute 'size'

这个问题有解决办法吗？是什么导致了这个错误？谢谢你的建议

Answer 1

让我们再看看那个错误。

AttributeError: 'NoneType' object has no attribute 'size'

因此，您（或 mtcnn）在代码的某处试图从 None 变量调用 size 属性。您正在使用以下命令将 frame 传递给 mtcnn：

 boxes,_ = mtcnn.detect(frame)

这正是您看到该错误的地方。因为您将 None 变量传递给 mtcnn。为了防止它，你可以在调用这个方法之前阻止它。换句话说：

ret, frame = capture.read()
if frame == None:
    continue

'NoneType' object has no attribute 'size' - 如何使用mtcnn进行人脸检测？

'NoneType' object has no attribute 'size' - how to face detection using mtcnn?

python

face-recognition

face-detection

neural-network

pytorch