Closed zhangyunming closed 3 years ago
The restore model needs pair inputs: (LR_image, parsing_map)
. You may try to combine them to one input and split them inside the forward function.
hi, thank you so much . i have change the two input into one, the reslut demo is same as two inputs
when i translate to onnx , it said:
the 'mv' is not suport by onnx 9 10 11 12.
I am not familiar with ONNX and did not consider the code compatibility with onnx. You may have to debug the error and rewrite some operators by yourself.
thanks , but i can not find the operator "mv", did you use the torch.mv in the code?
Most possible reason is that ONNX does not support spectral_norm
. You may try to find if there are any solution to that.
hi, thanks for the great work.
when i translate the psfrgan_latest_net_G.pth to onnx, there is a error ocuured, the parsing is ok.
can you give some advices. thanks