Skip to content

Commit 9868d72

Browse files
author
Clément Pinard
committed
update README and required packages
1 parent 2fd5646 commit 9868d72

File tree

2 files changed

+19
-10
lines changed

2 files changed

+19
-10
lines changed

README.md

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -26,32 +26,41 @@ These networks expect a BGR input in range `[-0.5,0.5]` (compared to RGB in pyto
2626

2727
## Prerequisite
2828

29-
pytorch >= 0.4.1
30-
tensorboard-pytorch
31-
tensorboardX >= 1.4
32-
scipy
33-
argparse
29+
```
30+
pytorch >= 0.4.1
31+
tensorboard-pytorch
32+
tensorboardX >= 1.4
33+
spatial-correlation-sampler>=0.0.8
34+
imageio
35+
argparse
36+
```
3437

3538
## Training on Flying Chair Dataset
3639

3740
First, you need to download the [the flying chair dataset](http://lmb.informatik.uni-freiburg.de/resources/datasets/FlyingChairs.en.html) . It is ~64GB big and we recommend you put it in a SSD Drive.
3841

3942
Default HyperParameters provided in `main.py` are the same as in the caffe training scripts.
4043

41-
Example usage for FlowNetS :
44+
* Example usage for FlowNetS :
4245

43-
python main.py /path/to/flying_chairs/ -b8 -j8 -a flownets
46+
```bash
47+
python main.py /path/to/flying_chairs/ -b8 -j8 -a flownets
48+
```
4449

4550
We recommend you set j (number of data threads) to high if you use DataAugmentation as to avoid data loading to slow the training.
4651

4752
For further help you can type
4853

49-
python main.py -h
54+
```bash
55+
python main.py -h
56+
```
5057

5158
## Visualizing training
5259
[Tensorboard-pytorch](https://github.com/lanpa/tensorboard-pytorch) is used for logging. To visualize result, simply type
5360

54-
tensorboard --logdir=/path/to/checkoints
61+
```bash
62+
tensorboard --logdir=/path/to/checkoints
63+
```
5564

5665
## Training results
5766

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
torch>=0.4.1
22
torchvision
33
numpy
4-
spatial-correlation-sampler
4+
spatial-correlation-sampler>=0.0.8
55
tensorboardX>=1.4
66
imageio
77
argparse

0 commit comments

Comments
 (0)