search

Linfeng-Tang / SwinFusion

This is official Pytorch implementation of "SwinFusion: Cross-domain Long-range Learning for General Image Fusion via Swin Transformer"
161 stars 20 forks source link

https://github.com/Linfeng-Tang/SwinFusion #5

Open jyyuan666 opened 1 year ago

jyyuan666 commented 1 year ago

您好，我直接下载您的MSRS数据集，并放置在 ./Dataset/trainsets/MSRS/

Download the training dataset from MSRS dataset, and put it in ./Dataset/trainsets/MSRS/.

之后，直接运行：python test_swinfusion.py --model_path=./Model/Infrared_Visible_Fusion/Infrared_Visible_Fusion/models/ --iter_number=10000 --dataset=MSRS --A_dir=IR --B_dir=VI_Y

报错：subprocess.CalledProcessError: Command '['/home/yuan3080/anaconda3/envs/Imagefusion_paper7/bin/python', '-u', 'main_train_swinfusion.py', '--local_rank=2', '--opt', 'options/swinir/train_swinfusion_vif.json', '--dist', 'True']' returned non-zero exit status 1.

请问我是哪一步没有操作吗？谢谢

Linfeng-Tang commented 1 year ago

你检查一下有没有VI_Y这个文件夹 没有的话要自己先用代码生成Y通道的可见光图像，如果还有问题的话，可以加我QQ：2458707789（备注学校+姓名），我可以利用向日葵尝试远程协助你解决这个问题哈

ChNanAn commented 1 year ago

最后解决了嘛，怎么解决的啊？

Linfeng-Tang commented 1 year ago

请问 你在运行程序的时候遇到问题了吗

ChNanAn commented 1 year ago

你好 我下载了VIS-NIR Scene dataset，解压之后是 nirscene1\9个不同种类的文件夹(0000_nir、 0000_rgb-0100_nir 、0100_rgb)nir与rgb的图像对， 我看项目下载的项目中有一个Dataset文件夹，还有一个Dataset的压缩包，我解压了Dataset压缩包，将解压后的数据集直接放入./Dataset/trainsets/Nirscene/， 运行python test_swinfusion.py --model_path=./Model/RGB_NIR_Fusion/RGB_NIR_Fusion/models/ --iter_number=10000 --dataset=NirScene --A_dir=NIR --B_dir=VI_Y命令，就出错了， 错误信息： 断言错误:./Dataset/testsets/NirScene\NIR没有有效的图像文件 我需要手动将数据集中RGB 图像跟NIR图像对分开存储在两个文件夹嘛？我是用的是windows系统，这个命令是不是需要修改路径格式 比如\？

Linfeng-Tang commented 1 year ago

你需要根据dataloader的需要将下载的数据集分别存储到NIR和VI_Y两个文件夹下，注意 近红外和可见光的Y通道图像需要保持一致的命名。 关于Windows的一些修改需要你自己尝试修改哈 我用的事Ubuntu系统训练的

ChNanAn @.***> 于2023年5月5日周五 15:12写道：

你好 我下载了VIS-NIR Scene dataset，解压之后是 nirscene1\9个不同种类的文件夹(0000_nir、 0000_rgb-0100_nir 、0100_rgb)nir与rgb的图像对，

我看项目下载的项目中有一个Dataset文件夹，还有一个Dataset的压缩包，我解压了Dataset压缩包，将解压后的数据集直接放入./Dataset/trainsets/Nirscene/， 运行python test_swinfusion.py --model_path=./Model/RGB_NIR_Fusion/RGB_NIR_Fusion/models/ --iter_number=10000 --dataset=NirScene --A_dir=NIR --B_dir=VI_Y命令，就出错了， 错误信息： 断言错误:./Dataset/testsets/NirScene\NIR没有有效的图像文件 我需要手动将数据集中RGB 图像跟NIR图像对分开存储在两个文件夹嘛？我是用的是windows系统，这个命令是不是需要修改路径格式 比如\？

— Reply to this email directly, view it on GitHub https://github.com/Linfeng-Tang/SwinFusion/issues/5#issuecomment-1535821510, or unsubscribe https://github.com/notifications/unsubscribe-auth/AOFH5INDHUTAG2NL6XUEVILXESR4HANCNFSM6AAAAAAUXPWJLQ . You are receiving this because you commented.Message ID: @.***>