mayavi 取消可视化界面并保存图片

如题目，这种代码需要是在生成 occ 真值时产生的

主要参考 surroundocc 的 tools/gen/ 下的 visual.py 文件；
在可视化时，这块代码即可插入循环也可单独显示*

# ---- 取消可视化界面 ----
# 使用 Offscreen Engine   
mlab.options.offscreen = True

voxel_size = 0.5
figure = mlab.figure(size=(1080, 1080), bgcolor=(1, 1, 1))
plt_plot_fov = mlab.points3d(
     fov_voxels[:, 0],
     fov_voxels[:, 1],
     fov_voxels[:, 2],
     fov_voxels[:, 3],
     colormap="viridis",
     # scale_factor=1,
     scale_factor=voxel_size - 0.05 * voxel_size,
     mode="cube",
     opacity=1.0,
     vmin=0,
     vmax=19,
 )

plt_plot_fov.glyph.scale_mode = "scale_by_vector"
plt_plot_fov.module_manager.scalar_lut_manager.lut.table = colors
save_seman_jpg = save_gt_seman_jpg_path + f'/occ_gt_semantic_jpg_{curr_idx}.jpg'
mlab.savefig(save_seman_jpg)
mlab.close()

# 恢复默认 Offscreen Engine 设置
mlab.options.offscreen = False