# create a Python 3.9.6 virtual environment
$ source <directory of virtual environment>/bin/activate
$ cd <this repository>
$ pip install -r requirements.txt
-
unittest
$ cd <this repository>/unittest $ python -m unittest
-
training
# modify the configuration in scripts/configs/train.yml $ python <this repository>/scripts/train.py \ <this repository>/scripts/configs/train.yml
-
fine-tune
# modify the configuration in scripts/configs/fine_tune.yml $ python <this repository>/scripts/fine_tune.py \ <this repository>/scripts/configs/fine_tune.yml
-
evaluation
$ python <this repository>/scripts/evaluate.py \ <checkpoint directory>/<name>.pth
-
qualitive results
$ python <this repository>/scripts/visualize.py \ <checkpoint directory>/best.val.pth \ --option <option> \ --output <directory for qualitive results>
options comments VL-CMU-CD aligned PSCD aligned VL-CMU-CD-diff_1 unaligned (adjacent distance == 1) VL-CMU-CD-diff_-1 unaligned (adjacent distance == -1) VL-CMU-CD-diff_2 unaligned (adjacent distance == 2) VL-CMU-CD-diff_-2 unaligned (adjacent distance == -2)
- TBD
- release source code
- release datasets module
- release models module
- release train/fine-tune/evaluation/visualize scripts
- release pretraining weight
- examples of inference on new scenes