A ideia central é reproduzir SOR e PU-Net-Pytorch, consulte o código de lyqun/PU-Net_pytorch , mas no meu ambiente experimental archote==1.9.0, a parte de calcular a distância emd está sempre errada, então use daerduoCarey /PyTorchEMD O módulo de cálculo de distância emd.
Deve-se observar que após ser processado pelo módulo SOR, a entrada de dados no PU-Net precisa ser processada.
Link do código: Kandy990125/PUNet-Pytorch