有没有办法下载我用python中的matplotlib pyplot库创建的图像？[副本]

Question

我觉得这可能是一个基本问题，但我需要为一个项目创建一个具有随机维度的椭圆数据库。我现在正在Google Colab中执行以下代码：

import numpy as np
import cv2
from matplotlib import pyplot as plt
import pandas as pd
from PIL import Image 

from skimage import color

for i in range(2):
  img = np.full((64,64,3), 255, dtype=np.uint8)
  center_x = np.random.randint(30,45)
  center_y = np.random.randint(30,45)
  major_axis = np.random.randint(1,9)
  minor_axis= np.random.randint(1,9)
  angle = 0
  B = 0
  G = 0
  R = 0
  ellipseCoords = [img, (center_x,center_y), (major_axis, minor_axis), angle, 0, 360, (B,G,R), -1]
  a = cv2.ellipse(img, (center_x, center_y), (major_axis, minor_axis), angle, 0, 360, (B, G, R), -1)
  plt.imshow(a)
  a.shape

上面的代码按照预期创建并绘制了两个椭圆。我最终需要大约64-100来完成最后的事情，所以单独保存图像似乎非常耗时。更糟糕的是，我会这么做，但我更喜欢用计算机化的方式。

这可能吗？我似乎找不到任何有帮助的东西。。。

谢谢

Answer 1

您可以在每次迭代结束时使用matplotlib的Savefig函数（https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.savefig.html），如下所示：

for i in range(n):
    # generate elipse

    plt.imshow(a)
    plt.savefig(f'elipse_{i}.png')