Awesome
CoBRa
The is the github page for ECCV submission ``Learning Counterfactual Thoughts for Bias-Robust Vision-Language Reasoning''
Updates
- Inital Update [2024/03/06]
Dataset Download
| Updated on | Questions and Annotations | Figures | Question Count | Figure Count |
|---|---|---|---|---|
| Mar 06, 2024 | CoBRa.csv | CoBRa.zip | 64604 | 14151 |
Environment Setup
The following softwares are needed
- Python >= 3.8
- Cuda >= 11.3
- Pytorch >= 12.0
- numpy, pyyaml, openai, opencv-python, pillow, tqdm, transformers
CoCT
Train the TLM
The code is developped on top of implementations of ``TLM''. Please check the repo for MLM+TLM parameters
python train_TLM.py
Prompt the LVLMs
check ./scripts