Analog_Med

This repository contains the code for training UNet, UNet++ and SwinUNet models on analog using AIHWKIT

Dataset:

The models are trained on the data-sceince-bowl-2018

• The code for other datasets will be added soon

Models:

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