
关于Vision Transformer的图像去雾算法研究及其实现(含Python代码和文档说明).zip
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本资源深入探讨了基于Vision Transformer的先进图像去雾技术,并提供了详细的Python实现代码与文档指导。通过结合深度学习模型的优势,该方案能够有效提升去雾效果,为计算机视觉领域提供新的研究视角和技术支持。
基于Vision Transformer的图像去雾算法研究与实现源码及文档说明.zip包含预处理数据的部分内容:将训练数据图像切分成大小为256*256的小图。下载的数据集存放在路径 /home/dell/桌面/TPAMI2022/Dehazing/#dataset/NH_haze/ 内,该目录下有两个文件夹:train 和 test。
对于训练数据集的处理使用命令:
```
python3 generate_patches_SIDD.py --src_dir /home/dell/桌面/TPAMI2022/Dehazing/#dataset/NH_haze/train --tar_dir /home/dell/桌面/2022毕业设计/Datasets/NH-HAZE/train_patches
```
训练代码为My_train.py,使用如下命令运行:
```
python3 ./My_train.py --arch Uformer --nepoch 270 --batch_size 32 --env My_Infor_CR --gpu 1 --train_ps 128 --train_dir /media/dell/fd6f6662-7e3
```
全部评论 (0)
还没有任何评论哟~


