PyTorch-libtorch-U-Net
PyTorch-libtorch-U-Net copied to clipboard
A customizable 1D/2D U-Net model for libtorch (PyTorch C++ UNet)
Results
3
PyTorch-libtorch-U-Net issues
Sort by
recently updated
recently updated
newest added
Would it be possible to offer a pytorch binding? So this could be called from Python code?
Dear team, thanks you for great project! I try write data loader in C++ for training, but not much success. Did you try this code on an open source DB?
good job
1
good job. It is much useful than torch::jit::trace