YOLOV5----修改损失函数-SE

主要修改yolo.py、yolov5s.yaml及添加SE.py

一、SE.py

import numpy as np
import torch
from torch import nn
from torch.nn import init



class SEAttention(nn.Module):

    def __init__(self, channel=512