La idea central es reproducir SOR y PU-Net-Pytorch, consulte el código de lyqun/PU-Net_pytorch , pero en mi entorno experimental torch==1.9.0, la parte de calcular la distancia emd siempre es incorrecta, así que use daerduoCarey/PyTorchEMD El módulo de cálculo de distancia emd.
Cabe señalar que después de ser procesado por el módulo SOR, la entrada de datos en PU-Net debe procesarse.
Enlace de código: Kandy990125/PUNet-Pytorch