Ajay Narayanan Sridhar (afs6372@psu.edu), Varun Darshana Parekh (vdp5094@psu.edu)
Folder structure:
GRTM/* - Code files for U-net-based architecture Restormer/* - Code files for Restormer architecture
Script to run the demo:
python Restormer/demo1.py --task Deraining
We can share the conda environment if you want to try training/testing.