Loading model using PyTorch Hub: pytorch.org/hub/mateuszbuda_brain-segmentation-pytorch_unet
import torch model = torch.hub.load('mateuszbuda/brain-segmentation-pytorch', 'unet', in_channels=3, out_channels=1, init_features=32, pretrained=True)
Dataset used for development and evaluation was made publicly available on Kaggle
TorchScript is a way to create serializable and optimizable models from PyTorch code. Any TorchScript program can be saved from a Python process and loaded in a process where there is no Python dependency.
Firstly run model_script.py to script python model and it can be load with both python and c++ without any python dependency
when you get scripted model you can use it with torchscript library in C++
brainUnet.cpp file gets path of scripted model and input image and return predicted mask
' mkdir build cd build cmake ..
'
