Function get_data

src/data/dataloader.py:176–211

(opt)

def get_data(opt):
    train_opt = opt["datasets"]["train"]
    val_opt = opt["datasets"]["val"]
    num_classes = opt["train"]["num_classes"]

    # train
    train_images, train_targets = get_paths(train_opt)
    val_images, val_targets = get_paths(val_opt)

    if train_opt["type"] == "Segmentation":
        train_data = LoadSegData(train_opt)
        trainDS = train_data.init(train_images, train_targets)

    elif train_opt["type"] == "Classification":
        train_data = LoadClassData(train_opt, num_classes)
        trainDS = train_data.init(train_images, train_targets)

    else:
        assert (
            False
        ), f"Found data type as {train_opt['type']} but it should be one of Segmentation/Classification"

    if val_opt["type"] == "Segmentation":
        val_data = LoadSegData(val_opt)
        valDS = val_data.init(val_images, val_targets)

    elif val_opt["type"] == "Classification":
        val_data = LoadClassData(val_opt, num_classes)
        valDS = val_data.init(val_images, val_targets)

    else:
        assert (
            False
        ), f"Found data type as {val_opt['type']} but it should be one of Segmentation/Classification"

    return trainDS, valDS