PyTorch-libtorch-U-Net icon indicating copy to clipboard operation
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. It is much useful than torch::jit::trace