Die zentrale Idee besteht darin, SOR und PU-Net-Pytorch zu reproduzieren, siehe Code von lyqun/PU-Net_pytorch , aber in meiner experimentellen Umgebung Torch==1.9.0 ist der Teil der Berechnung der emd-Entfernung immer falsch, also verwenden Sie daerduoCarey /PyTorchEMD Das Modul zur Berechnung der EMD-Distanz.
Es ist zu beachten, dass die in PU-Net eingegebenen Daten nach der Verarbeitung durch das SOR-Modul verarbeitet werden müssen.
Code-Link: Kandy990125/PUNet-Pytorch