From 241139fb3e365a7d5acb952b19d13e29fafb67d7 Mon Sep 17 00:00:00 2001 From: zjkjzj Date: Sat, 19 Sep 2020 10:02:11 +0800 Subject: [PATCH] =?UTF-8?q?docs(readme):=20=E6=9B=B4=E6=96=B0=E4=BD=BF?= =?UTF-8?q?=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 26 ++++++++++++++++++++++++-- 1 file changed, 24 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d33b5e8..f43e4dd 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ $ pip install -r requirements.txt ## 使用 -采用单`GPU`方式进行训练和测试,首先设置`GPU`和当前位置 +首先设置`GPU`和当前位置 ``` $ export CUDA_VISIBLE_DEVICES=1 @@ -53,8 +53,30 @@ $ export PYTHONPATH=. * 训练 ``` -# 训练HMDB51 +# 训练UCF101 +# 单GPU $ python tools/train.py --config_file=configs/tsn_resnet50_ucf101_rgb.yaml +# 多GPU +$ python tools/train.py \ +--config_file=configs/tsn_resnet50_ucf101_rgb.yaml \ +--eval_step=1000 \ +--save_step=1000 \ +-g=2 +``` + +* 测试 + +``` +# 单模态测试 +$ python tools/test.py +$ python tools/test.py configs/tsn_resnet50_ucf101_rgbdiff.yaml outputs/tsn_resnet50_ucf101_rgbdiff.pth +# 多模态融合测试 - RGB + RGBDiff +$ python tools/fusion.py +$ python tools/fusion.py \ +configs/tsn_resnet50_ucf101_rgb.yaml \ +outputs/tsn_resnet50_ucf101_rgb.pth \ +configs/tsn_resnet50_ucf101_rgbdiff.yaml \ +outputs/tsn_resnet50_ucf101_rgbdiff.pth ``` ## 主要维护人员