Commit Graph

14 Commits

Author SHA1 Message Date
zjut
97501d082b feat(net): 替换 token_mixer 为 SCSA
- 移除 Pooling 层,使用 SCSA(Spectral Correlation and Spatial Attention)层作为 token_mixer
- 更新模型架构,以适应新的 SCSA 层
- 修改测试脚本,针对特定数据集进行测试
2024-11-15 17:32:41 +08:00
zjut
4f805c2449 refactor(net): 重构网络结构并移除未使用的类
- 移除了未使用的导入语句
- 删除了多个未使用的类,包括 BaseFeatureExtraction1、BaseFeatureExtractionSAR等
- 重命名了部分类以更好地反映其功能,如将 BaseFeatureFusion 改为 BaseFeatureExtraction 等- 简化了部分代码结构,提高了代码的可读性和维护性
2024-11-15 11:07:25 +08:00
zjut
ece5f30c2d feat(componets): 新增多个图像处理模块
- 添加 DynamicFilter 模块,用于频域滤波和动态调整滤波器权重
- 添加 SCSA 模块,探索空间注意力和通道注意力之间的协同作用
- 添加 SMFA 模块,自调制特征聚合用于高效图像重建
- 添加 TIAM 模块,时空交互注意力用于变化检测
- 添加 UFFC 模块,无偏快速傅里叶卷积用于图像修复
- 更新 net.py,引入 SCSA 模块替换原有注意力机制
- 优化 train.py,调整导入模块以支持新功能
2024-11-15 09:28:49 +08:00
c023c0801d refactor(net): 注释掉 DetailFeatureExtraction、DetailFeatureFusion 和 DetailFeatureExtractionSAR 类中的 enhancement_module
- 在三个类中注释掉了 enhancement_module 的定义
- 该改动可能是为了暂时禁用增强模块的功能或进行调试
2024-11-15 09:14:15 +08:00
555515c2dc feat(net): 移除 WTConv2d,添加 DEConv 模块
- 删除了 WTConv2d 相关代码
- 新增了 DEConv 模块,包括多种卷积类型
- 更新了 net.py 中的相关调用,移除了 WTConv2d
2024-11-14 17:15:00 +08:00
30bbfdf86e feat(components): 添加 DEConv 和 SEBlock 组件
- 新增 DEConv 组件,用于细节增强卷积
- 新增 SEBlock组件,用于通道注意力机制
- 更新 net.py 中的 DetailNode 结构
- 调整 train.py 中的模型初始化
2024-11-14 16:59:11 +08:00
zjut
c1eed72f24 feat(net): 重构特征融合模块并添加新组件
- 新增 BaseFeatureFusion 和 DetailFeatureFusioin 类,用于特征融合
- 更新 ProjectRootManager 配置,使用本地 Python 3.8 环境
- 修改训练数据集路径
- 优化训练日志输出格式
2024-11-14 16:02:05 +08:00
5e561ab6f7 修改代码结构,提高可读性和可维护性;调整训练输出频率。
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
2024-10-09 11:35:06 +08:00
96ce7d5fda 修改代码结构,提高可读性和可维护性;调整训练输出频率。
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
2024-10-08 16:50:11 +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
f0765f0835
Update net.py 2024-06-29 10:54:33 +08:00
HXY13
da5da74611
Add files via upload 2024-06-03 19:36:29 +08:00