如何使用Python Google Colab将带有边框的图像存储在文件夹中?

艾丽莎·麦妮(Alisha Maini)

我试图在检测到人脸后保存图像。我尝试使用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")
Navan K.

所以代替

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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

如何通过Google colab读取所有图像的形状并将其显示在数据集文件夹中?

如何重命名文件夹 google colab 中的所有文件

如何从google colab提取文件夹(gdrive)中的所有.rar文件?

如何使用google colab运行导入自定义python模块(两个文件在存储库文件夹中的级别相同)的现有jupyter笔记本?

如何使用Apache POI将带有边框的图像添加到Word中的表格单元中?

如何将github存储库安装为Google colab中的当前目录文件夹?

使用 google colab 将相同类别的图像和 json 文件过滤到 google drive 中的另一个文件夹中

如何将带有 pip 文件的文件夹配置为本地 pip 存储库

使用带有Java的JSON API在Google Cloud Storage中插入对象(作为文件夹)

如何将具有相同名称的文件夹递归复制到Google Colab中的另一个文件夹

如何使用Google Colab python读取/循环文件夹中的多个.csv文件,然后将每个文件分配为函数参数

Google云端存储:如何在Python中(递归)删除文件夹

使用Python时如何在Google Colab中引用文件?

如何使用Google Cloud API获取给定存储桶中的文件夹列表

如何使用 Google Cloud Storage 将文件夹目标设置到存储桶中?

在Google Colab中使用桌面文件夹

如何在Google Colab中解压缩文件夹?

Python:如何使用folderId将文件上传到Google云端硬盘中的特定文件夹

Google云端存储:如何使用gsutil获取存储桶/文件夹中的新文件列表

如何使用Google的Python Drive Client API在Team Drive中创建文件夹?

如何将带有路线的Google Map转换为javascript中的图像

如何使用带有 Python 的 Google Drive API 覆盖文件?

如何使用带有 Python 的 Google Cloud Functions 将列表写入 Google Cloud Storage 中的文件

如何读取,写入和列出Google存储桶中的文件夹和文件?

如何使用blobstore和Google云存储将文件上传到AppEngine中的自定义文件夹

如何以编程方式重命名Google存储中的文件夹?

如何将 Google 应用引擎中的静态文件夹与存储桶相关联

使用Google Cloud Storage从Firebase存储中删除文件夹

如何在Xcode 6中将带有边框的按钮添加到故事板