README.md 1.4 KB
Newer Older
1
# BMN
L
lixuanyi 已提交
2 3 4
|config | pretrain | AR@100| AUC | gpu_mem(M) | iter time(s) | ckpt | log|
|-|-|-|-|-|-|- | -|
|bmn_400x100_9e_2x8_activitynet_feature | None |75.28|67.22|5420|3.27|[ckpt]()| [log]()|
5 6 7 8 9 10 11 12

### Data
1. Put the rescaled feature data folder `csv_mean_100` under `$MMACTION/data/activitynet_feature_cuhk/`

    The raw feature data could be found at [here](https://github.com/wzmsltw/BSN-boundary-sensitive-network)

2. Put the annotaion files under `$MMACTION/data/ActivityNet`

13
    The annotation files could be found at [here]()
14 15 16 17

### Checkpoint
1. Put the `tem_best.pth.tar` and `pem_best.pth.tar` under `checkpoints/`

18
    The ckpts could be found at [here]()
19 20 21 22 23 24 25

## Train
You can use the following command to train a model.
```Bash
python tools/train.py ${CONFIG_FILE} [optional arguments]

# Example: train BMN on ActivityNet features dataset
26
python tools/train.py config/localization/bmn/bmn_400x100_2x8_9e_activitynet_feature.py
27 28 29 30 31 32 33 34 35
```

## Test
You can use the following command to test a model.
```Bash
python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [optional arguments]

# Example: test BMN on ActivityNet feature dataset
# Note: If evaluated, then please make sure the annotation file for test data contains groundtruth.
36
python tools/test.py  config/localization/bmn/bmn_400x100_2x8_9e_activitynet_feature.py checkpoints/SOME_CHECKPOINT.pth --eval AR@AN --out results.json
37
```