zjut
|
97501d082b
|
feat(net): 替换 token_mixer 为 SCSA
- 移除 Pooling 层，使用 SCSA（Spectral Correlation and Spatial Attention）层作为 token_mixer
- 更新模型架构，以适应新的 SCSA 层
- 修改测试脚本，针对特定数据集进行测试
|
2024-11-15 17:32:41 +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 |
|
|
|
afd55abe9e
|
模型结构
DetailFeatureExtraction增加了一个增强残差
BaseFeatureExtraction增加了
x = self.WTConv2d(x)
|
2024-10-07 15:24:33 +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
|
d65393b3c4
|
Update test_IVF.py
|
2024-06-09 19:06:32 +08:00 |
|
HXY13
|
da5da74611
|
Add files via upload
|
2024-06-03 19:36:29 +08:00 |
|
