dongjoaquin commited on
Commit
5c85af8
·
verified ·
1 Parent(s): 0b65ac5

Upload MODEL_CARD.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. MODEL_CARD.md +68 -0
MODEL_CARD.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # RadioUNet 声学路径损耗预测模型
2
+
3
+ ## 模型描述
4
+
5
+ RadioUNet是一个基于U-Net架构的深度学习模型,用于快速准确地预测声学传播路径损耗。该模型可以在包含建筑物的真实传播环境中,从点声源到2D平面上所有点的路径损耗进行估计。
6
+
7
+ **应用场景**: 变电站噪声环境监测、声场实时计算
8
+
9
+ ## 模型架构
10
+
11
+ - **基础架构**: U-Net卷积神经网络
12
+ - **输入**:
13
+ - 建筑物/屏障布局图 (256×256)
14
+ - 声源位置图 (256×256)
15
+ - **输出**: 路径损耗热力图 (256×256)
16
+ - **参数量**: ~134MB
17
+
18
+ ## 训练数据
19
+
20
+ - **场景尺寸**: 512m × 512m
21
+ - **分辨率**: 2m/pixel
22
+ - **训练方法**: 基于RadioMapSeer数据集改编,针对声学场景优化
23
+
24
+ ## 性能指标
25
+
26
+ - **推理速度**: 比传统物理模拟快10-100倍
27
+ - **准确性**: 与物理仿真结果高度一致
28
+ - **GPU支持**: 支持CUDA加速,也可CPU运行
29
+
30
+ ## 使用方法
31
+
32
+ ```python
33
+ import torch
34
+ from lib.RadioUNet import RadioWNet
35
+
36
+ # 加载模型
37
+ model = RadioWNet(inputs=2, phase="firstU")
38
+ model.load_state_dict(torch.load('best_model.pth'))
39
+ model.eval()
40
+
41
+ # 推理
42
+ with torch.no_grad():
43
+ # input_tensor: [batch, 2, 256, 256]
44
+ # output: [batch, 1, 256, 256] 路径损耗热力图
45
+ output = model(input_tensor)
46
+ ```
47
+
48
+ ## 限制
49
+
50
+ - **2D限制**: 当前模型仅支持2D平面计算,忽略高度信息
51
+ - **固定尺寸**: 输入必须是512m×512m场景,映射到256×256像素
52
+ - **屏障敏感性**: 对屏障高度变化的敏感性有限
53
+
54
+ ## 引用
55
+
56
+ 基于论文: [RadioUNet: Fast Radio Map Estimation with Convolutional Neural Networks](https://arxiv.org/pdf/1911.09002.pdf)
57
+
58
+ ## 许可证
59
+
60
+ MIT License
61
+
62
+ ## 作者
63
+
64
+ 董季坤 (dongjoaquin)
65
+
66
+ ## 更新日期
67
+
68
+ 2026-03-12