|
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 |
|