YOLOv12即插即用-Pconv(风车卷积)

1.模块介绍

PinwheelConv(风车状卷积)充分利用了IRST(红外搜索与跟踪)中的高斯分布特性,以极少的参数实现了高效且更大感受野的特性。此外,本文还提出了一种简单而高效的 SD 损失函数,有效缓解了标签 IoU 变化带来的不稳定性。通过与现有卷积模块和损失函数的广泛对比,所提出的方法在准确性和鲁棒性方面始终优于最先进的方法,展现出卓越的性能和广泛的应用潜力。

2.改进后的结构图

YOLOv12即插即用-Pconv(风车卷积)_第1张图片

3.将PConv融合进YOLOv12

首先找到ultralytics下面的Conv文件夹新建PConv.py把下面代码写入:

import torch
import torch.nn as nn
import torch.nn.functional as F


def autopad(k, p=None, d=1):  # kernel, padding, dilation
    """Pad to 'same' shape outputs."""
    if d > 1:
        k = d * (k - 1) + 1 if

你可能感兴趣的:(YOLOv12改进,YOLO,深度学习,人工智能)