# image2image-baseline-model

**Repository Path**: ssfantasy/image2image-baseline-model

## Basic Information

- **Project Name**: image2image-baseline-model
- **Description**: pytorch version of medical image reconstruction baseline models (UNet and GAN)
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2020-01-14
- **Last Updated**: 2024-12-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# image2image-baseline-model
pytorch version of medical image reconstruction baseline models (UNet and GAN)

data_preparation.py: load from raw dicom/nifti data, augment images and save in pickle file.

loss.py: PSNR, SSIM, Perceptual Loss