matplotlib 有没有一种方法可以保存没有白色的图像?

yqlxgs2m  于 2022-12-27  发布在  其他
关注(0)|答案(2)|浏览(125)

关于图像中白色,我有两个问题。我正在尝试训练一个语义分割模型,但是我的数据目前是如何存储的。
首先,如果我使用plt.savefig(),图像会以白色细边框存储,如下所示:使用bbox_inches ='tight':

第二个问题是如何存储训练数据的模板。下面显示了车轮轨迹:

然而,为了使用这个遮罩层,它需要有单独的标签-这里是白色的,显示'background',应该只标记为0,红色应该标记为'1'.我不太知道如何做到这一点,谷歌搜索也没有带我去任何地方-主要问题是这个图像的背景(用plt.fill()生成)是白色的,我认为这是造成问题.

oprakyz7

oprakyz71#

关于你的第一个问题,阅读不同的其他问题和阅读savefig文档,我得到了这里:plt.savefig('您的图像. png ',边框英寸数='紧密',焊盘英寸数=0)

juud5qan

juud5qan2#

我认为你需要通过像素到像素的工作,并检查颜色是否是白色的尝试使用
从PIL导入图像r,b,g,p =图片. getpixel((0,0))打印("红色:{0},绿色:{1},蓝色:{2} ".格式(r,g,b))
在这里,您将了解该像素的RBG颜色,然后,您可以使用以下命令更改像素颜色
图像. putdata(数据、比例、偏移)
从左上角(0,0)开始,将像素值从序列对象复制到图像中。缩放和偏移值用于调整序列值:
像素=值 * 比例+偏移
如果省略比例,则默认为1.0。如果省略偏移,则默认为0.0。

相关问题