Skip to content

Commit bdeda17

Browse files
author
Dan Jia
committed
Update README
1 parent 737de21 commit bdeda17

File tree

1 file changed

+21
-18
lines changed

1 file changed

+21
-18
lines changed

README.md

Lines changed: 21 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -29,35 +29,38 @@ cd dr_spaam
2929
python setup.py install
3030
```
3131

32-
Download and put the [DROW dataset](https://github.com/VisualComputingInstitute/DROW) under `dr_spaam/data`.
33-
34-
Download the checkpoints from the release section and put them under `dr_spaam/ckpts`.
35-
36-
To measure the network inference time, run:
37-
```
38-
python bin/demo.py --time
39-
```
40-
41-
To visualize detections on an example sequence, run:
42-
```
43-
python bin/demo.py --dets
32+
Download and put the [DROW dataset](https://github.com/VisualComputingInstitute/DROW) under `dr_spaam/data`.
33+
Download the checkpoints from the [release section](https://github.com/VisualComputingInstitute/DR-SPAAM-Detector/releases) and put them under `dr_spaam/ckpts`.
34+
The directory should have the following layout.
4435
```
36+
dr_spaam
37+
├── data
38+
│ ├── DROWv2-data
39+
│ │ ├── test
40+
│ │ ├── train
41+
│ │ ├── val
42+
├── ckpts
43+
│ ├── drow_e40.pth
44+
│ ├── dr_spaam_e40.pth
45+
...
46+
```
4547

46-
To visualize detections with tracklets, run:
48+
Run `bin/demo.py` to measure the inference time (`--time`),
49+
to visualize detections on an example sequence (`--dets`),
50+
or to visualize tracklets (`--tracks`).
4751
```
48-
python bin/demo.py --tracks
52+
python bin/demo.py [--time/--dets/--tracks]
4953
```
5054

51-
To train your own network, run:
55+
To train a network, run:
5256
```
5357
python bin/train.py --cfg cfgs/dr_spaam.yaml
5458
```
5559

56-
To evaluat a checkpoint, run:
60+
To evaluat a checkpoint on the test set (on the validation set with `--val`), run:
5761
```
58-
python bin/eval.py --cfg cfgs/dr_spaam.yaml --ckpt ckpts/dr_spaam_e40.pth
62+
python bin/eval.py --cfg cfgs/dr_spaam.yaml --ckpt ckpts/dr_spaam_e40.pth [--val]
5963
```
60-
(add `--val` to evaluate on the validation set)
6164

6265
Integrating DR-SPAAM into other python projects is easy.
6366
Here's a minimum example.

0 commit comments

Comments
 (0)