主要修改yolo.py、yolov5s.yaml及添加ShuffleAttention.py
一、ShuffleAttention.py
import numpy as np
import torch
from torch import nn
from torch.nn import init
from torch.nn.parameter import Parameter
class ShuffleAttention(nn.Module):
def
主要修改yolo.py、yolov5s.yaml及添加ShuffleAttention.py
一、ShuffleAttention.py
import numpy as np
import torch
from torch import nn
from torch.nn import init
from torch.nn.parameter import Parameter
class ShuffleAttention(nn.Module):
def