The code runs on Python 3.8. You can create a virtualenv by running
virtualenv -p /usr/bin/python3.8 venv
source venv/bin/activate
pip install -r requirements.txt
- Login to wandb from your console
- To train LVD model:
python train_model.py --init_model 1 --ofm --dr 6 --pe 10 --pf 16 --pc --gain \
--trainset_dir <path_to_ReCRVD_dataset> \
--valset_dir <path_to_ReCRVD_dataset> \
--log_dir <path_to_dir_for_chkecpoints>
python eval_models.py
--data_dir <path_to_ReCRVD_dataset> \
--model <path_to_model>
ReCRVD dataset can be downloaded following instructions from the authors: https://github.com/cao-cong/RViDeformer