This repository contains the code for training UNet, UNet++ and SwinUNet models on analog using AIHWKIT
The models are trained on the data-sceince-bowl-2018
- The code for other datasets will be added soon
The models are taken from the monai package and preprocessing an training method taken from this repository
- UNet from monai. Paper: https://link.springer.com/chapter/10.1007/978-3-030-12029-0_40.
- NestedUNet(UNet++) from this repo code. Paper: https://arxiv.org/abs/1807.10165
- SWinUNETR from monai. Paper : https://arxiv.org/abs/2201.01266