4 个月前

ENet:一种用于实时语义分割的深度神经网络架构

ENet:一种用于实时语义分割的深度神经网络架构

摘要

在移动应用中,实现实时像素级语义分割的能力至关重要。近期针对这一任务的深度神经网络存在需要大量浮点运算且运行时间较长的问题,这限制了它们的实用性。本文提出了一种名为ENet(高效神经网络)的新型深度神经网络架构,专门设计用于低延迟操作的任务。ENet的速度最高可提升18倍,所需的浮点运算(FLOPs)减少75倍,参数量减少79倍,并且其精度与现有模型相当或更高。我们已在CamVid、Cityscapes和SUN数据集上对其进行了测试,并报告了与现有最先进方法的对比结果以及网络在精度与处理时间之间的权衡。我们还展示了所提出的架构在嵌入式系统上的性能测量结果,并提出了可能的软件改进措施以进一步提高ENet的速度。

代码仓库

Kaustav97/Colab
tf
GitHub 中提及
alalagong/LEDNet
pytorch
GitHub 中提及
TimoSaemann/ENet
caffe2
GitHub 中提及
PRBonn/bonnet
tf
GitHub 中提及
shanglianlm0525/CvPytorch
pytorch
GitHub 中提及
gjy3035/enet.pytorch
pytorch
GitHub 中提及
hustvl/yolop
pytorch
GitHub 中提及
e-lab/ENet-training
GitHub 中提及
fregu856/deeplabv3
pytorch
GitHub 中提及
zm0612/ENet-version-CPP
GitHub 中提及
davidtvs/PyTorch-ENet
pytorch
GitHub 中提及
sacmehta/ESPNet
pytorch
GitHub 中提及
m273033/masters_thesis
tf
GitHub 中提及
jiayuanwang-jw/yolov8-multi-task
pytorch
GitHub 中提及
Shun14/enet
paddle
GitHub 中提及
y-ouali/pytorch_segmentation
pytorch
GitHub 中提及
osmr/imgclsmob
mxnet
GitHub 中提及
adrshm91/segmentationEnet
tf
GitHub 中提及
gevero/enet_tensorflow
tf
GitHub 中提及
klintan/pytorch-lanenet
pytorch
GitHub 中提及
gtakos-ai/ENet
GitHub 中提及
soumik12345/Enet
pytorch
GitHub 中提及
kwotsin/TensorFlow-ENet
tf
GitHub 中提及
fregu856/segmentation
tf
GitHub 中提及
klintan/lanenet-pytorch
pytorch
GitHub 中提及
wutianyiRosun/CGNet
pytorch
GitHub 中提及
xiaoyufenfei/LEDNet
pytorch
GitHub 中提及
hustvl/YOLOP
pytorch
GitHub 中提及

基准测试

基准方法指标
real-time-semantic-segmentation-on-cityscapesENet
Frame (fps): 76.9
Time (ms): 13
mIoU: 58.3%
semantic-segmentation-on-cityscapesENet
Mean IoU (class): 58.3%
semantic-segmentation-on-scannetv2ENet
Mean IoU: 37.6%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供