Requirements pytorch torchvision numpy cv2 PIL matplotlib scikit-image scipy torchdiffeq Base code comes from https://github.com/DIAGNijmegen/neural-odes-segmentation