feat(actpred): 动作预测模块入库(PyTorch 训练/预测/ONNX 导出)

This commit is contained in:
qing committed 2026-08-08 06:25:06 +00:00
1 parent f93ae59521
commit 575c2a27a4
11 files changed
+752

No files matched your search

+13
View File
@@ -0,0 +1,13 @@
#!/bin/bash
cd "$(dirname "$0")" || exit
source ./venv/bin/activate
if [ "${1:-}" = "train" ]; then
shift
exec python3 train.py "$@"
elif [ "${1:-}" = "export" ]; then
shift
exec python3 export_onnx.py "$@"
else
exec python3 predict.py "$@"
fi