как сохранить рисунок в vis_bbox без белого фона, при построении с помощью matplotlib? - PullRequest
0 голосов
/ 12 марта 2020

Я пытаюсь сохранить изображение после прогнозирования vis_bbox с исходным размером изображения.

мой код:

from PIL import Image, ImageChops
import cv2

img = utils.read_image('/home/ubuntu/ui.jpg', color=True)
bboxes, labels,scores = model.predict([img])
bbox, label, score = bboxes[0], labels[0], scores[0],
colors = voc_colormap(label + 1)
bccd_labels = ('cell', 'cell')
vis_bbox(img, bbox,  label_names=bccd_labels, instance_colors=colors, alpha=0.9, linewidth=1.0)
plt.axis("off")


plt.savefig("/home/ubuntu/ins.jpg")

при сохранении изображение сохраняется с белым фоном и по умолчанию size (432 * 288).

Мне нужно сохранить прогнозное изображение из vis_bbox с исходным размером (1300 * 1300).

Любые предложения будут полезны!

...