A problem on test

[mochaojie]

I want to just use one image to show the result, how to do it?

[xinge008]

I do not release visualization code.

For a simple try, you can use the input data with batchsize=1, then you can get the result and follow other visualization code to get the final figure.