search

ClementPinard / SfmLearner-Pytorch

Pytorch version of SfmLearner from Tinghui Zhou et al.
MIT License
1.01k stars 224 forks source link

where is the model save？ #81

Closed yxh1993 closed 4 years ago

yxh1993 commented 4 years ago

when I input ''python3 run_inference.py --pretrained /path/to/dispnet --dataset-dir /path/pictures/dir --output-dir /path/to/output/dir'', I want to know where is the disnet and what is disnet mean.

And I also want to know what is dataset-dir mean and where is it.

ClementPinard commented 4 years ago

DispNet is the network used to compute the depth. I actually outputs the inverse of depth, which is in a stereo camera the disparity, hence the name. It is usually located in the checkpoints folder, but you can download the pretrained version here (from the README) and put it anywhere you like.

The dataset-dir option is a folder with pictutes in it. The inference script will then try to compute the depth map of each picture in the folder you will give it, and save the result as a numpy array or vizualisation picture in the folder /path/to/output/dir