मैं उनमें चेहरों का पता लगाने के बाद छवियों को बचाने की कोशिश कर रहा हूं। मैंने 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")
2
Alisha Maini 12 सितंबर 2020, 09:37

1 उत्तर

सबसे बढ़िया उत्तर

तो इसके बजाय

pixels = cv2.imread(filename)

आपको रखना चाहिए

pixels = pyplot.imread(filename)

जो मुझे लगता है कि एक वाक्यात्मक त्रुटि है।

इसे भी लूप से हटा दें

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')
1
Navan K. 12 सितंबर 2020, 09:55