错误代码:
writer.add_image("input", imgs, step, dataformats='HWC')
正确代码:
writer.add_images("input", imgs, step, dataformats='HWC')
错误原因:
add_image
只接收单一图像,而我们直接将imgs批量传入是不行的。
错误代码:
writer.add_image("input", imgs, step, dataformats='HWC')
正确代码:
writer.add_images("input", imgs, step, dataformats='HWC')
错误原因:
add_image
只接收单一图像,而我们直接将imgs批量传入是不行的。