Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -48,9 +48,7 @@ Five ground-roll intensity levels produce paired noisy / noise-label records:
|
|
| 48 |
|
| 49 |
## Model Architectures
|
| 50 |
|
| 51 |
-
- **
|
| 52 |
-
- **ResUNet-Plus** (`res_unet_plus`) — Wider ResUNet variant. Base channels: 64, depth: 4.
|
| 53 |
-
- **UNet-Plus** (`unet_plus`) — Wider UNet variant. Base channels: 64, depth: 4.
|
| 54 |
|
| 55 |
### Preprocessing
|
| 56 |
|
|
|
|
| 48 |
|
| 49 |
## Model Architectures
|
| 50 |
|
| 51 |
+
- **Enhanced Atten-UNet** (`enhanced_atten_unet`) — U-Net with residual blocks + attention-gated skip connections, trained with hybrid MSE + AFM (adaptive frequency modulation) loss. Predicts noise residual. Base channels: 32, depth: 4.
|
|
|
|
|
|
|
| 52 |
|
| 53 |
### Preprocessing
|
| 54 |
|