about INF

[Thatboy7]

excuse me . Is there an another method? the torch.diag methon isn't supported by onnx version and when i use torch.eye ,my tensorrt doesn‘t support it either.

[speedinghzl]

@Thatboy7 You could drop the INF if you train the model again.
The overlapping pixels have a minor effect on the final performance.