本文主要是介绍改进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 nnclass TransformerEncoderLayer
这篇关于改进YOLO系列 | YOLOv5/v7 引入高效的混合特征编码器 AIFI的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!