Baselines for PAT image enhancement and reconstruction

Related Works

Implemented	Architecture	Paper Title	Citation
✅	UNet	U-Net: Convolutional Networks for Biomedical Image Segmentation
✅	Uformer	Uformer: A General U-Shaped Transformer for Image Restoration
✅	UNet++	UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentation
✅	UNet 3+	UNet 3+: A Full-Scale Connected UNet for Medical Image Segmentation
	FD UNet	Fully Dense UNet for 2-D Sparse Photoacoustic Tomography Artifact Removal

Train

Run

torchrun --nproc_per_node=4 main.py --mode train --config configs/unet3p.py

Evaluation