cyd0806 commited on
Commit
d775861
·
verified ·
1 Parent(s): ee79125

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +55 -0
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - underwater-acoustic
5
+ - channel-estimation
6
+ - denoising
7
+ - deep-learning
8
+ ---
9
+
10
+ # DSS-Net Checkpoints
11
+
12
+ Pre-trained model checkpoints for **DSS-Net: Dynamic-Static Separation Networks for UWA Channel Denoising**.
13
+
14
+ ## Available Models
15
+
16
+ | Model | File | Size | Description |
17
+ |-------|------|------|-------------|
18
+ | **DSS-Net (Full)** | `dss_net_full_best.pth` | 499MB | Best performing model (NMSE: -25.27 dB) |
19
+ | Baseline U-Net | `baseline_unet_best.pth` | 355MB | Baseline for comparison (NMSE: -20.41 dB) |
20
+
21
+ ## Usage
22
+
23
+ ```python
24
+ import torch
25
+ from model import UNetDecomposer
26
+
27
+ # Load model
28
+ model = UNetDecomposer(
29
+ in_channels=2,
30
+ base_channels=64,
31
+ depth=4,
32
+ use_attention=True
33
+ )
34
+
35
+ # Load weights
36
+ checkpoint = torch.load('dss_net_full_best.pth', map_location='cpu')
37
+ model.load_state_dict(checkpoint['model_state_dict'])
38
+ model.eval()
39
+ ```
40
+
41
+ ## Citation
42
+
43
+ ```bibtex
44
+ @article{yang2025dssnet,
45
+ title={DSS-Net: Dynamic--Static Separation Networks for Physics-Inspired UWA Channel Denoising},
46
+ author={Yang, Xiaoyu and Chen, Yinda and Tong, Feng and Zhou, Yuehai},
47
+ journal={IEEE Transactions on Wireless Communications},
48
+ year={2025}
49
+ }
50
+ ```
51
+
52
+ ## Links
53
+
54
+ - **GitHub**: https://github.com/ydchen0806/dss_net
55
+ - **Paper**: IEEE TWC 2025