一、引言
最近在完成图像分析与理解的课程作业,当代码调试好后,准备运行,在最后保存图片部分却出现了问题,具体如下。
二、报错描述
问题代码
p = compute_map(args.input_0, args.input_1)
imageio.imsave("cm.png", (255 * p[0, :, :]).numpy())
报错截图
通过查询报错部分:TypeError: Cannot handle this data type: (1, 1), <i8,初步得知应该是数据类型部分的问题
打印变量p
查阅了大量的资料和阅读很多博文发现需要作出如下修改、
p = compute_map(args.input_0, args.input_1)
imageio.imsave("cm.png", np.uint8(p[0, :, :]*255))
再次运行程序即可、