whaifree/pfcfuse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
43 Commits 19 Branches 0 Tags 3.7 GiB
ad0b45e198
Commit Graph

43 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
zjut
ad0b45e198 feat(train): 添加模型输入功能- 在训练过程中增加了用户输入模型名称的功能 
- 打印出用户输入的模型名称
 2024-11-17 16:15:26 +08:00
zjut
231f42e924 refactor(net): 修改 DetailNode 类的默认参数 
- 将 DetailNode 类的 useBlock 参数默认值从 0 改为 1
- 这个修改可能会影响网络结构的选择和初始化

两分支wtconv
 2024-11-17 16:13:59 +08:00
zjut
b0030fe87f refactor(net): 修改 DetailNode 的使用方式 
- 在 DetailFeatureFusion 类中，将 DetailNode 的 useBlock 参数从 1 修改为 2
- 在 DetailFeatureExtraction 类中，将 DetailNode 的 useBlock 参数从 0 修改为 2
 2024-11-17 16:09:05 +08:00
zjut
260e3aa760 feat(net): 添加 WTConv2d 层并修改 DetailNode 使用- 在 net.py 中添加了 WTConv2d 层的导入- 修改了 DetailNode 类的构造函数，增加了 useBlock 参数 
- 根据 useBlock 参数的值，选择使用 WTConv2d层或 InvertedResidualBlock- 更新了 DetailFeatureFusion 和 DetailFeatureExtraction 类，指定了 DetailNode 的 useBlock 参数
 2024-11-17 15:57:41 +08:00
zjut
3b7b64c915 test(test_IVF.py): 更新模型路径和测试名称 
- 更新模型路径为新的权重文件
- 修改模型名称以反映新的增强方法
 2024-11-17 15:49:42 +08:00
zjut
0336fc23ba feat(net): 为 DetailNode模块添加可选卷积块 
- 在 DetailNode 类中引入 useBlock 参数，用于选择不同的卷积块
- 新增 DepthwiseSeparableConvBlock 类，实现深度可分离卷积
- 根据 useBlock 的值，选择使用 DepthwiseSeparableConvBlock 或 InvertedResidualBlock
- 优化了网络结构，提供了更多的灵活性和选择性
 2024-11-17 10:33:24 +08:00
zjut
a435e035c0 test: 更新测试模型并修改模型名称 
- 更新测试模型路径为 whaiFusion11-16-23-08.pth
- 修改模型名称为 "PFCFuse Enhance  SCSA"
 2024-11-17 09:53:12 +08:00
zjut
23293e9138 feat(net): 替换 token_mixer 为 SCSA 模块 
- 引入新的 SCSA（空间和通道协同注意力）模块
- 用 SCSA 替换原有的 Pooling层作为 token_mixer
- 删除了未使用的 SEBlock.py 文件- 移除了与当前项目无关的 TIAM(CV).py 文件
 2024-11-16 23:07:51 +08:00
zjut
aae81d97fd feat(net): 新增基础特征融合和细节特征融合模块 
- 添加了 BaseFeatureFusion 和 DetailFeatureFusion 两个新类
- 更新了 train.py 中的导入和实例化语句
 2024-11-16 21:39:34 +08:00
zjut
9224f9b640 test: 更新测试模型路径 
- 修改了 test_IVF.py 文件中的模型路径
- 将旧路径 whaiFusion11-15-22-09.pth 更改为新路径 whaiFusion11-16-11-20.pth
 2024-11-16 21:37:02 +08:00
zjut
e8a0212bbb feat(net): 增加 SAR 图像处理支持 
- 新增 BaseFeatureExtractionSAR 和 DetailFeatureExtractionSAR 模块
- 修改 DIDF_Encoder 类，支持 SAR 图像输入
- 更新测试和训练脚本，增加 SAR 图像处理相关逻辑
 2024-11-16 12:59:07 +08:00
zjut
0ef5760d76 test(test_IVF.py): 更新测试模型并调整测试数据集 
- 更新模型路径为 whaiFusion11-15-17-48.pth
- 将测试数据集从 ["TNO","RoadScene"] 修改为 ["sar"]- 修改模型名称为 "PFCFuse 最基本版本"
 2024-11-15 21:48:32 +08:00
zjut
344de69cb2 refactor(net): 重构网络结构并移除未使用的代码 
- 移除了未使用的导入语句和冗余代码
- 重构了某些类和方法，提高了代码可读性
- 删除了未使用的变量和注释掉的代码块
- 简化了部分代码结构，提高了运行效率
 2024-11-15 17:48:42 +08:00
whai
c023c0801d refactor(net): 注释掉 DetailFeatureExtraction、DetailFeatureFusion 和 DetailFeatureExtractionSAR 类中的 enhancement_module 
- 在三个类中注释掉了 enhancement_module 的定义
- 该改动可能是为了暂时禁用增强模块的功能或进行调试
 2024-11-15 09:14:15 +08:00
whai
555515c2dc feat(net): 移除 WTConv2d，添加 DEConv 模块 
- 删除了 WTConv2d 相关代码
- 新增了 DEConv 模块，包括多种卷积类型
- 更新了 net.py 中的相关调用，移除了 WTConv2d
 2024-11-14 17:15:00 +08:00
whai
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
whaifree
b6486dbaf4 添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。 2024-10-26 18:37:15 +08:00
whaifree
7d6d629786 添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。 2024-10-09 12:04:46 +08:00
whaifree
15eb10b512 添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。 2024-10-09 11:57:57 +08:00
whaifree
5e561ab6f7 修改代码结构，提高可读性和可维护性；调整训练输出频率。 
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
 2024-10-09 11:35:06 +08:00
whaifree
96ce7d5fda 修改代码结构，提高可读性和可维护性；调整训练输出频率。 
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
 2024-10-08 16:50:11 +08:00
whaifree
afd55abe9e 模型结构 
DetailFeatureExtraction增加了一个增强残差
BaseFeatureExtraction增加了
x = self.WTConv2d(x)
 2024-10-07 15:24:33 +08:00
zjut
af0a9f358c -- 2024-10-07 13:54:44 +08:00
whaifree
d65f6ec3ce 减少训练过程中的输出频率 
修改了训练脚本，将训练过程中的进度输出频率从每批输出改为每100批输出一次，以减少输出量，提高训练效率。
 2024-10-06 22:16:42 +08:00
whaifree
87fa4dfcc6 增加了INN部分的残差连接模块，修改了训练和测试代码以提高代码的可读性和可维护性。- 在train.py中添加了打印所有参数的代码，以方便检查和记录 2024-10-06 22:12:47 +08:00
whaifree
7068b627c4 增加了INN部分的残差连接模块，修改了训练和测试代码以提高代码的可读性和可维护性。- 在train.py中添加了打印所有参数的代码，以方便检查和记录 2024-10-06 16:58:13 +08:00
whaifree
faacea007c 修改代码实现，提高代码可读性和可维护性 2024-10-06 16:42:18 +08:00
whaifree
c9e054e236 修改代码实现，提高代码可读性和可维护性 2024-10-05 20:49:35 +08:00
HXY13
5e3fc11c37
Add files via upload 2024-09-19 10:31:39 +08:00
HXY13
f0765f0835
Update net.py 2024-06-29 10:54:33 +08:00
HXY13
282d38e52a
Update README.md 2024-06-29 10:53:11 +08:00
HXY13
8f50f55503
Update README.md 2024-06-26 15:32:37 +08:00
HXY13
6265086a58
Delete model directory 2024-06-09 19:13:33 +08:00
HXY13
d65393b3c4
Update test_IVF.py 2024-06-09 19:06:32 +08:00
HXY13
656c8ba0a1
Add files via upload 2024-06-05 10:48:04 +08:00
HXY13
b02b41f794
Add files via upload 2024-06-04 22:00:55 +08:00
HXY13
8b1957743f
Update README.md 2024-06-03 19:48:42 +08:00
HXY13
1ad1204210
Update README.md 2024-06-03 19:48:08 +08:00
HXY13
054bf829e7
Update README.md 2024-06-03 19:42:00 +08:00
HXY13
da5da74611
Add files via upload 2024-06-03 19:36:29 +08:00
HXY13
e6852193de
Update README.md 2024-05-01 15:46:07 +08:00
HXY13
1751361913
Initial commit 2024-05-01 15:44:50 +08:00