File tree 1 file changed +9
-7
lines changed
1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -4,13 +4,13 @@ This repo contains the *PyTorch* version of *MatchNet*.
4
4
5
5
## Dependencies
6
6
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
8
8
9
9
``` bash
10
10
pip install sk-video
11
11
```
12
12
13
- + [ tqdm] ( https://github.com/tqdm/tqdm ) : progress bar
13
+ + [ * tqdm* ] ( https://github.com/tqdm/tqdm ) : progress bar
14
14
15
15
``` bash
16
16
conda config --add channels conda-forge
@@ -20,12 +20,14 @@ conda install tqdm
20
20
21
21
## Train * MatchNet*
22
22
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.
28
28
29
29
``` bash
30
30
python -u main.py < CLI arguments> | tee train.log
31
31
```
32
+
33
+ To run on a specific GPU, say ` n ` , type ` CUDA_VISIBLE_DEVICES=n ` just before ` python ... ` .
You can’t perform that action at this time.
0 commit comments