available_models

available_models(task=Task.SEMANTIC_SEGMENTATION)

The architectures this package can build for a task.

Returns

Name Type Description
list[dict[str, Any]] One row per architecture: the name you write under architecture, the arguments the
list[dict[str, Any]] model block accepts, and what the architecture does differently.

sorted(row[“name”] for row in available_models()) [‘linknet’, ‘res_u_net’, ‘u_net’, ‘u_net_plus_plus’] [row[“name”] for row in available_models(“object_detection”)] [‘yolo3’] “separation” in available_models(“instance_segmentation”)[0][“arguments”] True