更新 generate_radar_data.py

This commit is contained in:
2025-12-05 03:17:13 +00:00
parent bc74e259df
commit 9ce94e8ad4

View File

@@ -50,4 +50,73 @@ def generate_radar_test_data(filename="Radar_Accuracy_Test_Data.csv"):
t_e = current_e + random.gauss(0, 0.005)
# --- B. 生成雷达测量值 (真值 + 传感器高斯噪声) ---
m_d = t_d + random.gauss(0, sigma_dist_km)
m_d = t_d + random.gauss(0, sigma_dist_km)
m_a = t_a + random.gauss(0, sigma_angle)
m_e = t_e + random.gauss(0, sigma_angle)
# --- C. 累加误差平方 (用于验证RMS) ---
sq_err_dist += (m_d - t_d)**2
sq_err_az += (m_a - t_a)**2
sq_err_el += (m_e - t_e)**2
# --- D. 格式化数据 (保留小数位) ---
# 距离保留3位小数(米级), 角度保留2位
row = {
"序号": i,
"真值_方位": f"{t_a:.2f}",
"真值_俯仰": f"{t_e:.2f}",
"真值_距离": f"{t_d:.3f}",
"测量_方位": f"{m_a:.2f}",
"测量_俯仰": f"{m_e:.2f}",
"测量_距离": f"{m_d:.3f}"
}
data_rows.append(row)
# 更新下一时刻状态
current_d += speed_km_s * time_interval_s
current_a += az_rate
current_e += el_rate
# ================= 3. 验证 RMS (控制台输出) =================
rms_dist = math.sqrt(sq_err_dist / num_points) * 1000 # 转换为米
rms_az = math.sqrt(sq_err_az / num_points)
rms_el = math.sqrt(sq_err_el / num_points)
print("-" * 50)
print(f"生成的模拟数据统计 (验证是否满足指标):")
print(f"距离 RMS: {rms_dist:.2f} m (指标 <= 10m) -> {'合格' if rms_dist<=10 else '不合格'}")
print(f"方位 RMS: {rms_az:.2f} ° (指标 <= 0.6°) -> {'合格' if rms_az<=0.6 else '不合格'}")
print(f"俯仰 RMS: {rms_el:.2f} ° (指标 <= 0.6°) -> {'合格' if rms_el<=0.6 else '不合格'}")
print("-" * 50)
# ================= 4. 导出为 CSV (Excel格式) =================
# 对应表A.3 的表头结构
headers = [
"搜索序号",
"真值-方位(度)", "真值-俯仰(度)", "真值-距离(km)",
"探测结果-方位(度)", "探测结果-俯仰(度)", "探测结果-距离(km)"
]
try:
with open(filename, mode='w', newline='', encoding='utf-8-sig') as f:
writer = csv.writer(f)
# 写入表头
writer.writerow(headers)
# 写入数据行
for row in data_rows:
writer.writerow([
row["序号"],
row["真值_方位"], row["真值_俯仰"], row["真值_距离"],
row["测量_方位"], row["测量_俯仰"], row["测量_距离"]
])
print(f"\n成功生成文件: {filename}")
print("您可以直接使用 Excel 打开该文件,并将数据复制到 word 报告的表A.3中。")
except PermissionError:
print(f"\n错误: 无法写入文件 {filename}。请检查文件是否正在被 Excel 打开。")
if __name__ == "__main__":
generate_radar_test_data()