Commit Graph

8 Commits

Author SHA1 Message Date
85dc7a92ed feat(net): 添加时空注意力机制- 在 net.py 中引入 SpatiotemporalAttentionFullNotWeightShared 模块
- 在 Restormer_Decoder 类中添加时空注意力机制处理基础特征和细节特征
2024-10-26 19:18:01 +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