Commit Graph

6 Commits

Author SHA1 Message Date
zjut
0e6064181a 新增SAR图像处理功能并优化模型性能
- 新增BaseFeatureExtractionSAR和DetailFeatureExtractionSAR类,专门用于SAR图像的特征提取
- 在Restormer_Encoder中加入SAR图像处理的支持,通过新增的SAR特征提取模块提高模型对SAR图像的处理能力
- 更新test_IVF.py,增加对SAR图像的测试,验证模型在不同数据集上的性能
- 通过这些修改,模型在TNO和RoadScene数据集上的表现得到显著提升,详细指标见日志文件
2024-10-09 12:09:30 +08:00
d65f6ec3ce 减少训练过程中的输出频率
修改了训练脚本,将训练过程中的进度输出频率从每批输出改为每100批输出一次,以减少输出量,提高训练效率。
2024-10-06 22:16:42 +08:00
7068b627c4 增加了INN部分的残差连接模块,修改了训练和测试代码以提高代码的可读性和可维护性。- 在train.py中添加了打印所有参数的代码,以方便检查和记录 2024-10-06 16:58:13 +08:00
faacea007c 修改代码实现,提高代码可读性和可维护性 2024-10-06 16:42:18 +08:00
c9e054e236 修改代码实现,提高代码可读性和可维护性 2024-10-05 20:49:35 +08:00
HXY13
da5da74611
Add files via upload 2024-06-03 19:36:29 +08:00