malar
/
abnormal-spatiotemporal-ae
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Minor fixes in README

pull/4/head
Shean 2017-09-18 00:18:29 +08:00
parent dd35b4b29b
commit c2a8568665
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -15,11 +15,11 @@ Prerequisites:
You can use the `Dockerfile` provided to build the environment then enter the environment using `nvidia-docker run --rm -it -v HOST_FOLDER:/share DOCKER_IMAGE bash`.
To train the model, just run `python start_train.py`. Default configuration can be found at `config.yml`. You need to prepare video dataset you plan to train/evaluate on. You may get the benchmark dataset videos from respective authors. For each dataset, put the training videos into `VIDEO_ROOT_PATH/DATASET_NAME/training_videos` and testing videos into `VIDEO_ROOT_PATH/DATASET_NAME/testing_videos`. Example structure of training videos for `avenue` dataset:
`VIDEO_ROOT_PATH/avenue/training_videos`
- `01.avi`
- `02.avi`
- ...
- `16.avi`
- `VIDEO_ROOT_PATH/avenue/training_videos`
- `01.avi`
- `02.avi`
- ...
- `16.avi`
Once you have trained the model, you may now run `python start_test.py` after setting the parameters at the beginning of the file.