HarDNet-MSEG 是全新的卷积神经网络,用于息肉分割。它包含主干和解码器。主干是低存储器流量卷积神经网络,称为 HarDNet68,它已成功应用于各种 CV 任务,包括图像分类、对象检测、多对象跟踪和语义分割。解码器部分的灵感源自于以快速准确的显着目标检测而闻名的“Cascaded Partial Decoder”(级联部分解码器)。下图显示了息肉分割的结果。
图 1. 息肉分割示例
下表列出了 Vitis AI Library 支持的息肉分割模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | HardNet_MSeg_pt | PyTorch |