divideconcept / PyTorch-libtorch-U-Net

A customizable 1D/2D U-Net model for libtorch (PyTorch C++ UNet)
MIT License
26 stars 4 forks source link