Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
210 stars 37 forks source link

关于train与test #37

Open CrystalD97 opened 11 months ago

CrystalD97 commented 11 months ago

大佬您好,我是这个方向的小白,正在学习过程。 想请问您该模型如何使用自己的图片进行融合呢,万分感谢

Linfeng-Tang commented 11 months ago

在test.py中指定ir_dir和vi_dir的路径即可。

CrystalD97 commented 11 months ago

在test.py中指定ir_dir和vi_dir的路径即可。

大佬您好,我尝试将test_imgs中的ir和vi图像替换成我自己的图,就没有再修改网络。 但是在运行时出现了FileNotFoundError: [Errno 2] No such file or directory: ./test_imgs/ir/vi (1).jpeg的错误 同时请教大佬在test之前是否需要对图像进行什么处理呢? 叨扰大佬,初次接触有很多问题不太懂,期待大佬回复,感谢!

Linfeng-Tang commented 11 months ago

需要保持ir_dir和vi_dir中文件名和后缀都是一致的呢! 如果有任何问题可以QQ联系:2458707789

CrystalD97 commented 11 months ago

需要保持ir_dir和vi_dir中文件名和后缀都是一致的呢! 如果有任何问题可以QQ联系:2458707789

谢谢大佬!已经跑通模型了!感谢!

oyeoye123 commented 4 months ago

跑通了,可是训练完缺失一个参数文件