def get_input_wh(model_name):
    """Get input_width and input_height of the model."""
    yolo_dim = model_name.split('-')[-1]
    if 'x' in yolo_dim:
        dim_split = yolo_dim.split('x')
        if len(dim_split) != 2:
            raise ValueError('ERROR: bad yolo_dim (%s)!' % yolo_dim)
        w, h = int(dim_split[0]), int(dim_split[1])
    else:
        h = w = int(yolo_dim)
    if h % 32 != 0 or w % 32 != 0:
        raise ValueError('ERROR: bad yolo_dim (%s)!' % yolo_dim)
    return w, h