【导读】今天给大家推荐一个很棒的开源项目,该项目主要汇总了一系列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
更多详细的模型结构、模型训练与测试信息见:https://github.com/murufeng/FUIR
推荐阅读
西电IEEE Fellow团队出品!最新《Transformer视觉表征学习全面综述》
如何做好科研?这份《科研阅读、写作与报告》PPT,手把手教你做科研
最新 2022「深度学习视觉注意力 」研究概述,包括50种注意力机制和方法!
【重磅】斯坦福李飞飞《注意力与Transformer》总结,84页ppt开放下载!
分层级联Transformer!苏黎世联邦提出TransCNN: 显著降低了计算/空间复杂度!
清华姚班教师劝退文:读博,你真的想好了吗?
2021李宏毅老师最新40节机器学习课程!附课件+视频资料
欢迎大家加入DLer-计算机视觉技术交流群!
大家好,群里会第一时间发布计算机视觉方向的前沿论文解读和交流分享,主要方向有:图像分类、Transformer、目标检测、目标跟踪、点云与语义分割、GAN、超分辨率、人脸检测与识别、动作行为与时空运动、模型压缩和量化剪枝、迁移学习、人体姿态估计等内容。
进群请备注:研究方向+学校/公司+昵称(如图像分类+上交+小明)
长按识别,邀请您进群!