主要修改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 Parameterclass ShuffleAttention(nn.Module):def