Pytorch implementation of Refinenet network.
Note that the NYU Depth_v2 dataset is already include in this repository, the only thing have to change is in
config.py
Simply train the RefineNet:
python train.py
Also make sure that you have already have all the settings done in config.py
.
Simply test the RefineNet:
python test.py
- Add more datasets.
- Provide pretrained-model
- ...
Thanks markshih91 for providing these easy-to-use codes.