我试图在检测到人脸后保存图像。我尝试使用matplotlib.pyplot.savefig(img,bbox_inches ='tight',pad_inches = 0)和此存储的图像,但存储的所有图像都是空白的白色图像。我得到的输出
我想要存储的是:-我期望的输出
#face detection with mtcnn on a photograph
from matplotlib import pyplot
from mtcnn.mtcnn import MTCNN
from matplotlib.patches import Rectangle
from matplotlib.patches import Circle
import glob
import cv2
import matplotlib
# draw an image with detected objects
def draw_image_with_boxes(filename, result_list):
# load the image
data = pyplot.imread(filename)
# plot the image
pyplot.imshow(data)
# get the context for drawing boxes
ax = pyplot.gca()
# plot each box
for result in result_list:
# get coordinates
x, y, width, height = result['box']
# create the shape
rect = Rectangle((x, y), width, height, fill=False, color='red')
# draw the box
ax.add_patch(rect)
# draw the dots on eyes nose ..
#for key, value in result['keypoints'].items():
# create and draw dot
#dot = Circle(value, radius=2, color='red')
#ax.add_patch(dot)
# show the plot
pyplot.show()
#filename = '/content/drive/My Drive/images/*.jpg'
i = 1
for filename in glob.glob('/content/drive/My Drive/images/*.jpg'):
pixels = cv2.imread(filename)
# load image from file
#pixels = pyplot.imread(filename)
# create the detector, using default weights
detector = MTCNN()
# detect faces in the image
faces = detector.detect_faces(pixels)
# display faces on the original image
new = draw_image_with_boxes(filename, faces)
img = '/content/drive/My Drive/boundingBox' + '/image_' + str(i) + '.jpg'
matplotlib.pyplot.savefig(img, bbox_inches='tight', pad_inches=0)
i+=1
print("Done")
所以代替
pixels = cv2.imread(filename)
你应该把
pixels = pyplot.imread(filename)
我猜这是一个语法错误。
还要从for循环中删除它
matplotlib.pyplot.savefig(img, bbox_inches='tight', pad_inches=0)
然后在定义的函数中draw_image_with_boxes(filename, result_list)
,在“显示绘图”注释之后,添加以下代码行。
pyplot.savefig('/content/drive/My Drive/TESTBBOX/' + 'image'+ str(i)+'.jpg',bbox_inches='tight')
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句