This repository is bult in Pytorch 2.0.1 and tested on Ubuntu 20.04.4 LTS5.4.0-148-generic environment (Python 3.8, CUDA 11.7, cuDNN 8.5). Follow these instructions:
- Clone our repository:
git clone https://github.com/hellopipu/PromptMR.git
cd PromptMR
- Create conda environment The Conda environment used can be recreated using the env.yaml file:
conda env create -f env.yaml
For CMRxRecon dataset, follow the instruction here.
For FastMRI knee dataset, follow the instruction here.