# RCAN_TensorFlow_for_NPU

**Repository Path**: monster_w/rcan_tensorflow_npu

## Basic Information

- **Project Name**: RCAN_TensorFlow_for_NPU
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-12-11
- **Last Updated**: 2025-01-02

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# RCAN_TensorFlow_for_NPU

#### 项目说明
本项目是RCAN的TensorFlow版本在昇腾NPU实现的代码。    
参考论文
[Image Super-Resolution Using Very Deep Residual Channel Attention Networks](https://arxiv.org/abs/1807.02758)    
参考实现
https://github.com/dongheehand/RCAN-tf

#### Requirement
tensorflow==1.15.0    
cann==5.0.2    
numpy==1.17.5    
scikit-image==0.17.2    

#### 默认配置
- 网络结构    
RCAN×2    
- 训练超参（单卡）    
batch_size:32    
learning_rate(LR):2*1e-4    
max_steps:1e6    
#### 安装教程
使用git clone {repository url}    
或者下载本项目到本地并解压缩

#### 数据集
- [DIV2K](https://data.vision.ee.ethz.ch/cvl/DIV2K/)
- [Benchmarks](https://cv.snu.ac.kr/research/EDSR/benchmark.tar)
#### 使用说明
- 数据集准备    
在项目的工作目录下，在命令行执行如下脚本一步完成数据集下载和更改名称
    ```
    bash dl_datasets.sh
    ```
- 训练和推理
在完成数据集准备工作之后，在命令行执行如下命令进行训练和训练后的推理，得到精度最好的模型
    ```
    bash train_testcase.sh
    ```