dataset/__init__.py

# from .UAV.dataset import UavMapPair
from .dataset import UavMapDatasetModule

# modules = {"UAV": UavMapPair}