projects/sa2va/datasets/__init__.py

from .data_utils import sa2va_collect_fn, sa2va_collect_fn_multitask

# Training-only dataset imports are omitted for inference/demo deployment.
# from .glamm_refcoco import Sa2VARefCOCODataset
# from .glamm_entityseg import Sa2VAEntitySegDataset
# from .glamm_material import Sa2VAMaterialDataset