改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子

CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子

改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子_第1张图片

论文地址:https://arxiv.org/pdf/2103.06255v2.pdf
代码地址:https://github.com/d-li14/involution


Involution卷积,文章描述说它比convolution更轻量更高效,形式上比self-attention更加简洁,可以在各种视觉任务的模型上取得精度和效率的双重提升。
改进YOLO系列 | CVPR 2021 | Involution:超越convolution和self-attention的神经网络算子_第2张图片


内卷积代码

import torch
from torch import nn

# -----

你可能感兴趣的:(YOLOv5/v7进阶实战,神经网络,深度学习,python)