File size: 804 Bytes
907ff09 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# Copyright (c) Facebook, Inc. and its affiliates.
import logging
import os
from detectron2.data.datasets.builtin_meta import _get_builtin_metadata
from detectron2.data.datasets.lvis import get_lvis_instances_meta
from .lvis_v1 import custom_register_lvis_instances
_CUSTOM_SPLITS = {
"cc3m_v1_val": ("cc3m/validation/", "cc3m/val_image_info.json"),
"cc3m_v1_train": ("cc3m/training/", "cc3m/train_image_info.json"),
"cc3m_v1_train_tags": ("cc3m/training/", "cc3m/train_image_info_tags.json"),
}
for key, (image_root, json_file) in _CUSTOM_SPLITS.items():
custom_register_lvis_instances(
key,
get_lvis_instances_meta('lvis_v1'),
os.path.join("datasets", json_file) if "://" not in json_file else json_file,
os.path.join("datasets", image_root),
)
|