def create_model_dict(input_channels, num_classes=[4, 1], name='2d'):
    """
        Creates `model_dict` dictionary from parameters.
        Parameters
        ----------
        input_channels: int
            1 indicates gray-channle image, 3 indicates RGB image.
        num_classes: list
            [4, 1] -> 4 indicates offset in x, offset in y, margin in x, margin in y; 1 indicates seediness score
        name: string
    """
    model_dict = {
        'name': 'branched_erfnet' if name=='2d' else 'branched_erfnet_3d',
        'kwargs': {
            'num_classes': num_classes,
            'input_channels': input_channels,
        }
    }
    print(
        "`model_dict` dictionary successfully created with: \n -- num of classes equal to {}, \n -- input channels equal to {}, \n -- name equal to {}".format(
            input_channels, num_classes, name))
    return model_dict