可以单独提供一个用于测试新的两张输入图片的脚本吗? 我已经训练好了模型,并且在最后一步infer_vis出来的验证图片效果不错  但是缺乏一个可以单独调用模型用于测试的脚本,请问作者有什么解决办法或者可以提供一下代码吗