About inference

[szxmachinelearning]

Great job. I'm very interested in your work. But I have a question. The inference code given can only use one image, what should I do if I want to batch inference images? If I only change the batch size, it will cause a dimension error.

[kangnn]

Do you know how to inference？I meet some problem.