Skip to content

Commit 93c4f65

Browse files
committed
Improve main README.md
1 parent 7e547e7 commit 93c4f65

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,13 @@ This repo contains the *PyTorch* version of *MatchNet*.
44

55
## Dependencies
66

7-
+ [scikit-video](https://github.com/scikit-video/scikit-video): accessing images / videos
7+
+ [*scikit-video*](https://github.com/scikit-video/scikit-video): accessing images / videos
88

99
```bash
1010
pip install sk-video
1111
```
1212

13-
+ [tqdm](https://github.com/tqdm/tqdm): progress bar
13+
+ [*tqdm*](https://github.com/tqdm/tqdm): progress bar
1414

1515
```bash
1616
conda config --add channels conda-forge
@@ -20,12 +20,14 @@ conda install tqdm
2020

2121
## Train *MatchNet*
2222

23-
+ [`resize_and_split.sh`](data/resize_and_split.sh): Prepares your (video) data for training.
24-
Resizes videos present in folders of folders (*i.e.* directory of classes) and may split them into training and validation set.
25-
May also skip short videos and trim longer ones.
26-
+ [`main.py`](main.py): Script to start training.
27-
Use `-h` to print the command line arguments help.
23+
+ [`utils/resize_and_split.sh`](data/resize_and_split.sh): Prepares your (video) data for training.
24+
Resizes videos present in folders of folders (*i.e.* directory of classes) and may split them into training and validation set.
25+
May also skip short videos and trim longer ones.
26+
+ [`main.py`](main.py): Script to start training.
27+
Use `-h` to print the command line arguments help.
2828

2929
```bash
3030
python -u main.py <CLI arguments> | tee train.log
3131
```
32+
33+
To run on a specific GPU, say `n`, type `CUDA_VISIBLE_DEVICES=n` just before `python ...`.

0 commit comments

Comments
 (0)