Skip to content

Commit

Permalink
comment on additional flags
Browse files Browse the repository at this point in the history
Signed-off-by: fred@pegasus <[email protected]>
  • Loading branch information
fredzzhang committed Dec 5, 2021
1 parent 757876e commit 41e58fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ git submodule update
```

## Inference
We have implemented inference utilities with different visualisation options. Provided you have downloaded the model weights to `checkpoints/`, run the following command to visualise detected instances together with the attention maps from the cooperative and competitive layers.
We have implemented inference utilities with different visualisation options. Provided you have downloaded the model weights to `checkpoints/`, run the following command to visualise detected instances together with the attention maps from the cooperative and competitive layers. Use the flag `--index` to select images, and `--box-score-thresh` to modify the filtering threshold on object boxes.
```bash
python inference.py --resume checkpoints/upt-r50-hicodet.pt --index 8789
```
Expand Down

0 comments on commit 41e58fb

Please sign in to comment.