search

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
207 stars 37 forks source link

融合 #45

Open AlleyOop23 opened 5 months ago

AlleyOop23 commented 5 months ago

怎样只训练融合部分的模型

Linfeng-Tang commented 5 months ago

修改dataloader 避免加载分割标签 在train_fusion 函数中取消语语义损失 然后再main函数中只调用train_fusion函数即可。

AlleyOop23 commented 5 months ago

感谢！

AlleyOop23 commented 5 months ago

请问model文件夹中，三个权重文件分别是什么