Pytorch-UNet
Pytorch-UNet copied to clipboard
Add cpp inference project.(Msnhnet Engine)
1.Add pytorch to msnhnet tool . 2.Add cmake project. 3.CPU avx2 mode is supported. 4.GPU cuda cudnn mode is supported. 5.GPU cuda cudnn fp16 mode is supported. Check out #286.