基于VisionTransformer的图像去雾算法研究与实现python源码+项目说明+数据集.zip

资源内容介绍

基于VisionTransformer的图像去雾算法研究与实现python源码+项目说明+数据集.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/。内容来