For each dataset, use the provided preprocessing script for each of the train, validation, and test scripts as follows:
python preprocess.py --split train
python preprocess.py --split val
python preprocess.py --split test
For COCO2014, the dataset needs to be first downloaded using download.sh.
The scripts provided here build upon scripts from stevenstalder/NN-Explainer.