# FSRCNN

**Repository Path**: sn202011060/fsrcnn

## Basic Information

- **Project Name**: FSRCNN
- **Description**: 用pytorch实现FSRCNN
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2024-12-05
- **Last Updated**: 2024-12-05

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

upscale_factor = 2
训练集为BSDS500， 验证集为91-img
学习速率为1e-4

baby        FSRCNN PSNR:38.27626398438051
bird        FSRCNN PSNR:41.09826992341736
butterfly   FSRCNN PSNR:33.505489717366885
head        FSRCNN PSNR:35.612806026888904
woman       FSRCNN PSNR:35.3552475633922

set5上平均   FSRCNN PSNR:36.76961544308917

文件用途:
model_FSRCNN.py
    搭建神经网络

prepare_FSRCNN.py
    准备下采样后的图片

train_FSRCNN.py
    训练模型，并将训练好的模型命名为‘FSRCNN_训练轮数(epoch)’保存在本文件夹中
    将训练过程中的loss记录在文件夹logs_FSRCNN中，可以在terminal中调用 tensorboard --logdir=‘./FSRCNN/logs_FSRCNN’查看

test_FSRCNN.py
    对训练好的模型进行测试
    含有PSNR计算函数 对模型进行评估

utils.py
    rgb image 转ycbcr  ycbcr转rgb

imresize.py
    resize函数 可以用其他库的resize进行代替