In Computational Imaging, Vision, and Language (CIVIL) @ IEEE International Conference on Data Science and Advanced Analytics 2024 (CIVIL @ DSAA 2024)

Please first specify TRAIN_DIR, VAL_DIR and SAVE_DIR in section TRAINING in config.yml

For single GPU training:

python train.py

For multiple GPUs training:

accelerate config
accelerate launch train.py

If you have difficulties on the usage of accelerate, please refer to Accelerate.

Please first specify TRAIN_DIR, VAL_DIR and SAVE_DIR in section TESTING in traning.yml

python test.py

This work was supported by the Science and Technology Development Fund, Macau SAR, under Grant 0141/2023/RIA2 and 0193/2023/RIA3.

If you find our work helpful for your research, please cite: