SuperGradients 允许用户训练用于语义分割任务的模型。该库包含预先训练的模型,例如 Cityscapes PPLiteSeg 模型,并提供用于加载自定义数据集的简单界面。
模型动物园
SuperGradients 包括各种用于语义分割任务的预训练模型。
Model Name | Dataset | IoU | Training Recipe | Resolution |
---|---|---|---|---|
DDRNet 23 | Cityscapes | 80.26 | cityscapes_ddrnet.yaml | [1024, 2048] |
DDRNet 23 Slim | Cityscapes | 78.01 |