修复小问题一枚，增加输出图片的函数

Signed-off-by: Meathill <meathill@gmail.com>