4 个月前

实例分割的路径聚合网络

实例分割的路径聚合网络

摘要

信息在神经网络中的传播方式至关重要。本文提出了一种路径聚合网络(Path Aggregation Network,简称PANet),旨在增强基于提议的实例分割框架中的信息流。具体而言,我们通过自底向上的路径增强,在较低层中加入精确的定位信号,从而缩短了低层特征与顶层特征之间的信息路径。我们引入了自适应特征池化技术,该技术将特征网格与所有特征层级连接起来,使得每个特征层级中的有用信息可以直接传递到后续的提议子网络。此外,我们创建了一个互补分支,以捕捉每个提议的不同视角,进一步提高掩码预测的准确性。这些改进易于实现,并且计算开销较小。我们的PANet在COCO 2017挑战赛的实例分割任务中获得了第一名,在目标检测任务中获得了第二名(无需大规模批量训练)。同时,PANet在MVD和Cityscapes数据集上也达到了最先进的水平。代码可在https://github.com/ShuLiu1993/PANet 获取。

代码仓库

ShuLiu1993/PANet
官方
pytorch
GitHub 中提及
YuefeiZ/PANet
tf
GitHub 中提及
texasinstruments/edgeai-yolox
pytorch
GitHub 中提及
texasinstruments/edgeai-yolov5
pytorch
GitHub 中提及
ultralytics/yolov5
pytorch
GitHub 中提及
wslerry/yolov5
pytorch
GitHub 中提及

基准测试

基准方法指标
instance-segmentation-on-cocoPANet
mask AP: 42.0
instance-segmentation-on-coco-minivalPANet (ResNet-50)
mask AP: 37.8
instance-segmentation-on-isaidPANet
Average Precision: 34.17
object-detection-on-cocoPANet (ResNeXt-101, multi-scale)
AP50: 67.2
AP75: 51.8
APL: 60.0
APM: 51.7
APS: 30.1
Hardware Burden:
Operations per network pass:
box mAP: 47.4
object-detection-on-isaidPANet
Average Precision: 41.66

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
实例分割的路径聚合网络 | 论文 | HyperAI超神经