改进YOLO系列 | YOLOv5/v7 引入高效的混合特征编码器 AIFI

论文地址：https://arxiv.org/abs/2304.08069

代码地址：https://github.com/PaddlePaddle/PaddleDetection

中文翻译：https://blog.csdn.net/weixin_43694096/article/details/131353118

---

注意！这个模块需要 torch>=1.9 才能使用

源代码

import torch
import torch.nn as nn


class TransformerEncoderLayer