zjut
|
8383c68ff7
|
Merge remote-tracking branch 'origin/base' into base_增加WTconv
# Conflicts:
# test_IVF.py
|
2024-11-18 09:39:41 +08:00 |
|
zjut
|
6738c9057d
|
feat(train): 添加模型保存路径打印功能
- 优化模型保存逻辑，将保存路径存储在变量中
- 在保存模型后，打印模型的保存路径
- 这个改动可以帮助用户更容易地找到和管理训练好的模型文件
|
2024-11-18 09:29:10 +08:00 |
|
zjut
|
ad0b45e198
|
feat(train): 添加模型输入功能- 在训练过程中增加了用户输入模型名称的功能
- 打印出用户输入的模型名称
|
2024-11-17 16:15:26 +08:00 |
|
zjut
|
aae81d97fd
|
feat(net): 新增基础特征融合和细节特征融合模块
- 添加了 BaseFeatureFusion 和 DetailFeatureFusion 两个新类
- 更新了 train.py 中的导入和实例化语句
|
2024-11-16 21:39:34 +08:00 |
|
zjut
|
e8a0212bbb
|
feat(net): 增加 SAR 图像处理支持
- 新增 BaseFeatureExtractionSAR 和 DetailFeatureExtractionSAR 模块
- 修改 DIDF_Encoder 类，支持 SAR 图像输入
- 更新测试和训练脚本，增加 SAR 图像处理相关逻辑
|
2024-11-16 12:59:07 +08:00 |
|
zjut
|
344de69cb2
|
refactor(net): 重构网络结构并移除未使用的代码
- 移除了未使用的导入语句和冗余代码
- 重构了某些类和方法，提高了代码可读性
- 删除了未使用的变量和注释掉的代码块
- 简化了部分代码结构，提高了运行效率
|
2024-11-15 17:48:42 +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 |
|
|
|
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 |
|
