From 65ac13074eaaa2447c644a2df71969d8a3dd1fae Mon Sep 17 00:00:00 2001 From: JYChen Date: Wed, 20 Apr 2022 10:43:57 +0800 Subject: [PATCH] add training link for action model, test=document_fix (#5755) --- deploy/pphuman/docs/action.md | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/deploy/pphuman/docs/action.md b/deploy/pphuman/docs/action.md index 82d1ac0e3..320252eae 100644 --- a/deploy/pphuman/docs/action.md +++ b/deploy/pphuman/docs/action.md @@ -65,6 +65,14 @@ python deploy/pphuman/pipeline.py --config deploy/pphuman/config/infer_cfg.yml \ ``` 4. 行为识别模型使用了[ST-GCN](https://arxiv.org/abs/1801.07455)，并基于[PaddleVideo](https://github.com/PaddlePaddle/PaddleVideo/blob/develop/docs/zh-CN/model_zoo/recognition/stgcn.md)套件完成模型训练。 +## 自定义模型训练 +我们已经提供了检测/跟踪、关键点识别以及识别摔倒动作的预训练模型，可直接下载使用。如果希望使用自定义场景数据训练，或是对模型进行优化，根据具体模型，分别参考下面的链接： +| 任务 | 算法 | 模型训练及导出文档 | +| ---- | ---- | -------- | +| 行人检测/跟踪 | PP-YOLOE | [使用教程](../../../configs/ppyoloe/README_cn.md#使用教程) | +| 关键点识别 | HRNet | [使用教程](../../../configs/keypoint#3训练与测试) | +| 行为识别 | ST-GCN | [使用教程](https://github.com/PaddlePaddle/PaddleVideo/tree/develop/applications/PPHuman) | + ## 参考文献 ``` @inproceedings{stgcn2018aaai, -- GitLab