改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子
Posted 迪菲赫尔曼
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子相关的知识,希望对你有一定的参考价值。
CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子
论文地址:https://arxiv.org/pdf/2103.06255v2.pdf
代码地址:https://github.com/d-li14/involution
Involution
卷积,文章描述说它比convolution
更轻量更高效,形式上比self-attention
更加简洁,可以在各种视觉任务的模型上取得精度和效率的双重提升。
内卷积代码
import torch
from torch import nn
# -----
以上是关于改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子的主要内容,如果未能解决你的问题,请参考以下文章
YOLOX: Exceeding YOLO Series in 2021 --- Study Notes