Skip to content

README.zh-CN.md

Latest commit

 

History

History
65 lines (49 loc) · 2.3 KB

README.zh-CN.md

File metadata and controls

65 lines (49 loc) · 2.3 KB
Raw

English · 中文 · Polski · Español · 日本語 · 한국어 · Русский · Français · Deutsch

Collision Detection AI

基于加速度计数据的碰撞检测，使用轻量级 PyTorch 模型（1.3k 参数），专为实时窗口推理而设计。

模型可视化

简单设置

python -m venv .venv && source .venv/bin/activate
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu132
pip install -r requirements.txt
collision-detection all          # 训练（生成数据、训练、评估）
collision-detection evaluate     # 在全部数据集上测试训练好的模型
collision-detection visualize    # 可视化训练模型的测试结果

所有命令都会在 data/（数据集）和 artifacts/（检查点、指标、图表——包括 artifacts/plots/visualization.png）中生成产物。

使用你自己的数据时，请先将 CSV 文件放入 data/raw/，然后在 collision-detection train 之前运行 collision-detection preprocess。

引用

如果本仓库对你的研究有帮助，欢迎引用：

APA 格式

Brzustowicz, S. (2026). Collision-Detection-AI: Collision detection from accelerometer data using a lightweight PyTorch model (Version 1.0.0) [Source code]. https://github.com/sebastianbrzustowicz/Collision-Detection-AI

BibTeX

@software{brzustowicz_collision_detection_ai_2026,
  author = {Sebastian Brzustowicz},
  title = {Collision-Detection-AI: Collision detection from accelerometer data using a lightweight PyTorch model},
  url = {https://github.com/sebastianbrzustowicz/Collision-Detection-AI},
  version = {1.0.0},
  year = {2026}
}

Tip

你还可以使用侧边栏中的 "Cite this repository" 按钮自动复制这些引用，或下载原始元数据文件。

许可证

CC BY-NC-ND 4.0 — 参见 LICENSE.md。

作者

Sebastian Brzustowicz Se.Brzustowicz@gmail.com