license:apache-2.0 region:us
---
license: apache-2.0
---

This repo provides a script to load refcoco, refcocog, and refcoco+ with local files. You should download the data first.

```bash
[DATA_ROOT]
|_ images
  |_ train2014
    |_ ... (images)
|_ refcoco
  |_ instances.json
  |_ refs(google).p
  |_ refs(unc).p
|_ refcocog
  |_ instances.json
  |_ refs(google).p
  |_ refs(umd).p
|_ refcoco+
  |_ instances.json
  |_ refs(unc).p
```

Usage:
```python
REPO_NAME = "yiqun/referit"
dataset = "refcoco"
DATA_ROOT = ...
split = "train"  # choices: train, val, test, testA, testB
split_by = "unc"
datasets.load_dataset(REPO_NAME, dataset, data_dir=DATA_ROOT, split=split, split_by=split_by)
```
