
from .jde import JointDataset, BallDataset, TilingBallDataset, HRDataset

def get_dataset(task, hr_version = False):
    """
    当前仅仅支持ball和player的检测任务、mot任务 返回联合数据集
    """
    if task == "ball":
        return TilingBallDataset
        # return BallDataset
    elif task == 'mot':
        if hr_version:
            return HRDataset
        return JointDataset
    else:
        raise NotImplementedError


from .jde import JointDataset, BallDataset, TilingBallDataset, HRDataset

def get_dataset(task, hr_version = False):
    """
    当前仅仅支持ball和player的检测任务、mot任务 返回联合数据集
    """
    if task == "ball":
        return TilingBallDataset
        # return BallDataset
    elif task == 'mot':
        if hr_version:
            return HRDataset
        return JointDataset
    else:
        raise NotImplementedError

当前仅仅支持ball和player的检测任务、mot任务 返回联合数据集 # return BallDataset