def get_input_node(model):
    """ return the model input node """
    iter = model.GetNodes()
    while iter.IsValid():
        node = iter.Get()
        if "InputNode" in node.GetRuntimeTypeName():
            return node
        iter.Next()
    return None