zjut
|
fa8106838e
|
feat(model): 重构模型并添加新功能
- 重新组织了模型结构，增加了新的特征融合模块
- 添加了深度可分离卷积块和新的细节特征提取模块
- 更新了数据处理流程，使用了新的数据集路径
- 调整了训练参数，增加了训练轮次和学习率- 优化了损失函数，使用了Huber损失替代MSE损失
|
2024-11-12 10:37:56 +08:00 |
|
zjut
|
e1a339e04b
|
feat(net): 替换 SMFA 为 SCSA 并调整相关配置
- 将 SMFA 模块替换为 SCSA 模块
- 更新项目配置，使用本地 Python 3.8 环境
-调整 SCSA 模块参数，如维度、头数等
- 优化注意力机制，提高模型性能
|
2024-11-08 12:04:52 +08:00 |
|
zjut
|
41b2ea1ff9
|
feat(net): 为 net.py 添加新的组件引用并优化前向传播逻辑
- 在 net.py 中引入 SMFA 组件
- 优化 BasicLayer 类的前向传播逻辑
- 添加 SMFA、DynamicFilter 和 UFFC 组件的实现
- 使用SMFA替代Pooling
self.WTConv2d = WTConv2d(dim, dim)
self.norm1 = LayerNorm(dim, 'WithBias')
self.token_mixer = SMFA(dim=dim)
# self.token_mixer = Pooling(kernel_size=pool_size) # vits是msa，MLPs是mlp，这个用pool来替代
self.norm2 = LayerNorm(dim, 'WithBias')
mlp_hidden_dim = int(dim * mlp_ratio)
self.poolmlp = PoolMlp(in_features=dim, hidden_features=mlp_hidden_dim,
act_layer=act_layer, drop=drop)
|
2024-11-05 14:09:59 +08:00 |
|
|
|
85dc7a92ed
|
feat(net): 添加时空注意力机制- 在 net.py 中引入 SpatiotemporalAttentionFullNotWeightShared 模块
- 在 Restormer_Decoder 类中添加时空注意力机制处理基础特征和细节特征
|
2024-10-26 19:18:01 +08:00 |
|
zjut
|
f4b3a933bf
|
train:调整训练日志输出频率
- 将训练日志的输出频率从每100 个批次调整为每个批次
- 此修改可以提供更频繁的训练进度反馈，便于实时监控训练状态
|
2024-10-26 19:04:12 +08:00 |
|
|
|
b6486dbaf4
|
添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。
|
2024-10-26 18:37:15 +08:00 |
|
|
|
7d6d629786
|
添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。
|
2024-10-09 12:04:46 +08:00 |
|
|
|
15eb10b512
|
添加 .idea/ 和 status.md到 .gitignore 文件，避免个人配置和状态文件被跟踪。在测试脚本中移除了不必要的打印语句。新增了测试日志和成功运行的日志文件。
|
2024-10-09 11:57:57 +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 |
|
|
|
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 |
|
|
|
d65f6ec3ce
|
减少训练过程中的输出频率
修改了训练脚本，将训练过程中的进度输出频率从每批输出改为每100批输出一次，以减少输出量，提高训练效率。
|
2024-10-06 22:16:42 +08:00 |
|
|
|
87fa4dfcc6
|
增加了INN部分的残差连接模块，修改了训练和测试代码以提高代码的可读性和可维护性。- 在train.py中添加了打印所有参数的代码，以方便检查和记录
|
2024-10-06 22:12:47 +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
|
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 |
|
