关于pytorch版推理代码的问题

[super-song-sir]

作者大大，可以在后续的更新中给出pytorch版的图片可视化推理代码吗？新手小白运行完训练代码想看一下推理的图片结果，但是自己的水平写不出来推理代码。非常感谢

[lyuwenyu]

see #95

[super-song-sir]

see #95

Sorry to bother you again, but do I have to transform the trained weights file through onnx before I can do inference? I didn't find any code for direct inference using pytorch. Could you please follow up with code that uses pytorch trained .pth files for direct visual test?

[super-song-sir]

见#95

Thank you,I find it at #136