gen_data.py

# gcc
# from vlmo.utils.write_conceptual_caption import make_arrow
# root = "/parastor/home/chenych/MML/data/gcc"
# arrows_root = "/parastor/home/chenych/MML/data/gcc/arrows_root"

# nlvr2
# from vlmo.utils.write_nlvr2 import make_arrow
# root = "/home/data/nlvr-master/"
# arrows_root = "/home/data/nlvr-master/images/arrows_root"

# coco
from vlmo.utils.write_coco_karpathy import make_arrow
root = "/home/data/coco2014/"
arrows_root = "/home/data/coco2014/"

if __name__ == "__main__":
    make_arrow(root, arrows_root)