Awesome
Requirements
- Python 3.6 (Anaconda is recommended)
- skimage
- imageio
- Pytorch 1.4.0
- torchvision 0.5.0
- tqdm
- pandas
- cv2 (pip install opencv-python)
Test
Prepare test data
- Download the KITTI2012 dataset and put folders
testing/colored_0
and testing/colored_1
in data/test/KITTI2012/original
- Cd to
data/test
and run generate_testset.m
to generate test data.
- (optional) You can also download KITTI2015, Middlebury or other stereo datasets and prepare test data in
data/test
as below:
datasets
└── test
├── dataset_1
├── hr
├── scene_1
├── hr0.png
└── hr1.png
├── ...
└── scene_M
└── lr_x4
├── scene_1
├── lr0.png
└── lr1.png
├── ...
└── scene_M
├── ...
└── dataset_N
Demo
python cvcnet_test.py
Train
python cvcnet_train.py