diff --git a/README.md b/README.md index dda8bdf..5b0e017 100644 --- a/README.md +++ b/README.md @@ -138,7 +138,7 @@ chmod ug+x download_middlebury_2014.sh && ./download_middlebury_2014.sh ``` and run ```Shell -python train_stereo.py --train_datasets middlebury_2014 --num_steps 4000 --image_size 384 1000 --restore_ckpt models/raftstereo-sceneflow.pth --batch_size 2 --train_iters 22 --valid_iters 32 --spatial_scale -0.2 0.4 --saturation_range 0 1.4 --n_downsample 2 --mixed_precision +python train_stereo.py --train_datasets middlebury_2014 --num_steps 4000 --image_size 384 1000 --lr 0.00002 --restore_ckpt models/raftstereo-sceneflow.pth --batch_size 2 --train_iters 22 --valid_iters 32 --spatial_scale -0.2 0.4 --saturation_range 0 1.4 --n_downsample 2 --mixed_precision ``` ## (Optional) Faster Implementation diff --git a/download_middlebury_2014.sh b/download_middlebury_2014.sh index fce26a3..009c8c2 100755 --- a/download_middlebury_2014.sh +++ b/download_middlebury_2014.sh @@ -1,4 +1,4 @@ -mkdir datasets/Middlebury/2014 +mkdir datasets/Middlebury/2014 -p cd datasets/Middlebury/2014 wget https://vision.middlebury.edu/stereo/data/scenes2014/zip/Adirondack-imperfect.zip wget https://vision.middlebury.edu/stereo/data/scenes2014/zip/Adirondack-perfect.zip