Question

import numpy as np 
import cv2
from matplotlib import pyplot as plt

img = cv2.imread('AB.jpg')
mask = np.zeros(img.shape[:2] , np.uint8) 

bgdModel = np.zeros((1,65), np.float64)
fgdModel = np.zeros((1,65), np.float64)

rect = (300 , 120 , 470 , 350)

#this modifies mask
cv2.grabCut(img,mask,rect,bgdModel, fgdModel , 5 , cv2.GC_INIT_WITH_RECT)

#If mask==2 or mask==1 , mask2 get 0, otherwise it gets 1 as 'uint8' type
mask2 = np.where((mask==2) | (mask==0),0,1).astype('uint8')

#adding additional dimension for rgb to the mask, by default it gets 1
#multiply with input image to get the segmented image
img_cut = img*mask2[: , : , np.newaxis]

plt.subplot(211),plt.imshow(img)
plt.title('Input Image') , plt.xticks([]),plt.yticks([])
plt.subplot(212),plt.imshow(img_cut)
plt.title('Grab cut'), plt.xticks([]),plt.yticks([])
plt.show()

编译时

我收到此错误：

python img.py AB.jpg
Traceback (most recent call last):
File "img.py", line 6, in <module>
mask = np.zeros(img.shape[:2] , np.uint8) 
AttributeError: 'NoneType' object has no attribute 'shape'

Answer 1

首先

python img.py AB.jpg

将无法正常工作（从当前目录加载AB.jpg）。要加载的文件在提供的示例的第6行中进行了硬编码：要按预期工作，它应该是这样的：

import sys
img = cv2.imread(sys.argv[1])

返回错误是因为正在从（当前工作目录）运行目录img.py中不存在AB.jpg，并且在尝试读取之前没有验证丢失的文件。

Answer 2

回答，因为社区把它带回来了。添加我的两个对象（美分）。

您看到该错误的唯一原因是您尝试获取信息或对首先不存在的对象执行操作。要检查，请尝试打印对象。喜欢，添加 -

print img      # such as this case
print contours # if you are working with contours and cant draw one
print frame    # if you are working with videos and it doesn't show

给你一个无。这意味着你没有正确阅读它。您提供的图像名称不存在或其路径错误。如果你发现这样的错误，那就是快速的事情 -

检查路径或将图像带到工作目录
检查您提供的名称是否正确（包括extension- .jpg，.png等）
将整个代码放在if语句中，对象和代码将继续（如果为true）
如果在评论中建议，将添加更多内容。

AttributeError：'NoneType'对象没有属性'shape'

2 个答案: