我试图在 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")
保存时，它以白色背景和默认大小（432 * 288）保存图像。
我需要使用原始尺寸（1300 * 1300）保存来自 vis_bbox 的预测图像。
任何的意见都将会有帮助！