【导读】
今天给大家推荐一个很棒的开源项目,该项目主要汇总了一系列Low-Level-vision领域当前主流的SOTA算法网络结构,主要包括图像去噪,图像去雨、图像去模糊等。该项目详细介绍了各大主流Low-Level-Vison数据集(如SIDD、GoPro、REDS、Rain13K等)的下载链接和训练配置教程,每个主流的算法网络结构都配备训练和测试命令方法。小伙伴们,快去Star关注吧!
项目地址:https://github.com/murufeng/FUIR
项目介绍
当前,对于Low-level-Vision中的图像恢复任务(Image Super-resolution, Image Denoising, Image Deblurring等) 已经出现了很多优秀实用的工具包。但是,对于去年出现的一系列Transformer网络架构还没有一个统一的项目将它集成起来,本项目主要在具体图像恢复任务的数据处理和网络训练配置上将当前主流的CNN网络和基于Transformer的网络架构集成了起来。本项目将作为图像恢复任务的一个灵活统一的工具。利用本项目可以快速实现图像去噪,图像/视频去模糊,图像去雨等一系列经典任务的训练。本项目提供在GOPRO、SIDD、REDS、Rain13K数据集上的数据预处理以及网络架构训练教程,后续我们将持续更新。希望本项目既能让图像处理初学者快速入门,又能服务科研和工业社区。
项目安装命令:
-
Star and Clone our repository
git clone https://github.com/murufeng/FUIR.git
cd FUIR
-
Make conda environment
conda create -n pytorch1.9.0_FUIR python=3.7 conda activate pytorch1.9.0_FUIR
-
Install dependencies
conda install pytorch=1.9.0 torchvision cudatoolkit=11.2 -c pytorch pip install matplotlib scikit-learn scikit-image opencv-python yacs joblib natsort h5py tqdm pip install einops gdown addict future lmdb numpy pyyaml requests scipy tb-nightly yapf lpip
-
Build
python setup.py develop --no_cuda_ext
数据集下载与配置:
-
图像去噪数据集 SIDD
-
图像去模糊数据集 GoPro
-
图像去模糊数据集REDS
-
图像去雨数据集 Rain13K
Model Zoo
更多详细的模型结构、模型训练与测试信息见:
https://github.com/murufeng/FUIR
推荐阅读
欢迎大家加入DLer-
计算机视觉技术交流
群!
大家好,群里会第一时间发布
计算机视觉方向的前沿论文解读和交流分享
,主要方向有:
图像分类、Transformer、目标检测、目标跟踪、点云与语义分割、GAN、超分辨率、人脸检测与识别、动作行为与时空运动、模型压缩和量化剪枝、迁移学习、人体姿态估计
等内容。
进群请备注:
研究方向+学校/公司+昵称
(如图像分类
+上交+小明)
👆
长按识别,邀请您进群!