Skip to content

Commit

Permalink
docs(readme): 更新安装和使用
Browse files Browse the repository at this point in the history
  • Loading branch information
zjykzj committed Sep 12, 2020
1 parent 42147bc commit c1126f0
Showing 1 changed file with 36 additions and 0 deletions.
36 changes: 36 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@

- [内容列表](#内容列表)
- [背景](#背景)
- [安装](#安装)
- [使用](#使用)
- [TODO](#todo)
- [主要维护人员](#主要维护人员)
- [致谢](#致谢)
Expand All @@ -30,6 +32,40 @@

[Temporal Segment Networks: Towards Good Practices for Deep Action Recognition](https://arxiv.org/abs/1608.00859)是视频分类任务中的经典实现

## 安装

通过requirements.txt安装运行所需依赖

```
$ pip install -r requirements.txt
```

处理数据时需要额外安装[denseflow](https://github.com/open-mmlab/denseflow),可以在[innerlee/setup](https://github.com/innerlee/setup)中找到安装脚本

## 使用

采用单`GPU`方式进行训练和测试,首先设置`GPU`和当前位置

```
$ export CUDA_VISIBLE_DEVICES=1
$ export PYTHONPATH=.
```

* 训练

```
# 训练HMDB51
$ python tools/train.py --config_file=configs/tsn_resnet50_hmdb51_rgb.yaml
# 恢复训练
$ python tools/train.py --config_file=configs/tsn_resnet50_hmdb51_rgb.yaml --resume
```

* 测试

```
```

## TODO

- [ ] 视频模式测试
Expand Down

0 comments on commit c1126f0

Please sign in to comment.