我正在尝试使用mayavi场景的透明度保存屏幕截图.基于this discussion,我编写了一个脚本来保存图形,但是结果是输出混乱.在最小的工作示例下面：在mayavi中,我创建一个具有两个球体的场景,然后先将其保存为RGB,然后再保存为RGBA格式.当RGBA文件混乱时,保存为RGB可以工作.我相信,问题出在mayavi方面,因为如果我从mayavi拍摄RGB图像,添加透明通道并使用PIL保存文件,结果将是我所期望的.
这是bug还是mayavi rgba格式应以某种方式转换为PIL接受的格式？
from mayavi import mlab
from PIL import Image
fig=mlab.figure(1, bgcolor=(1, 1, 1), size=(700, 700))
# Set camera position and properties
fig.scene.parallel_projection = True
fig.scene.show_axes = True
# Draw atoms
x, y, z, t = [0.0,1.0] , [0.0,1.0], [0.0,0.0], [1,2]
dat = mlab.pipeline.scalar_scatter(x, y, z, t, figure=fig)
fig = mlab.pipeline.glyph(dat,scale_mode='none', scale_factor=0.5, figure=fig)
imgmap_RGB = mlab.screenshot(figure=fig, mode='rgb', antialiased=True)
img_RGB = Image.fromarray(imgmap_RGB, 'RGB')
img_RGB.save('foo_RGB.png')
imgmap_RGBA = mlab.screenshot(figure=fig, mode='rgba', antialiased=True)
img_RGBA = Image.fromarray(imgmap_RGBA, 'RGBA')
img_RGBA.save('foo_RGBA.png')
mlab.show()
解决方法:
由于我不知道的某些原因,mayavi对于RGBA数据将返回0到1之间的浮点数,对于RGB数据将返回无符号整数,请参见https://github.com/enthought/mayavi/blob/master/mayavi/tools/figure.py#L304(我在文档中找不到信息).
要进行转换,请将img_RGBA = …行替换为
img_RGBA = Image.fromarray(np.array(imgmap_RGBA*255, dtype=np.uint8))
之后,我可以成功查看png文件.
标签：mayavi,python 来源： https://codeday.me/bug/20191026/1934150.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。