上传者: 42097508
|
上传时间: 2021-11-18 10:23:13
|
文件大小: 894KB
|
文件类型: -
快速语义分割
该存储库旨在为PyTorch中的移动设备提供准确的实时语义分段代码,并在Cityscapes上提供预训练的权重。 这可用于在各种现实世界的街道图像上进行有效的分割,包括Mapillary Vistas,KITTI和CamVid等数据集。
from fastseg import MobileV3Large
model = MobileV3Large . from_pretrained (). cuda (). eval ()
model . predict ( images )
这些模型是MobileNetV3 (大型和小型变体)的实现,具有基于LR- ASPP的修改后的细分头。 顶级型号在Cityscapes val上能够达到72.3%的mIoU精度,而在GPU上以高达37.3 FPS的速度运行。 请参阅下面的详细基准。
当前,您可以执行以下操作:
加载预训练的Mo