网页设计网站多少钱百度一下百度搜索百度一下
主要修改yolo.py、yolov5s.yaml及添加SE.py
一、SE.py
import numpy as np
import torch
from torch import nn
from torch.nn import initclass SEAttention(nn.Module):def __init__(self, channel=512
主要修改yolo.py、yolov5s.yaml及添加SE.py
一、SE.py
import numpy as np
import torch
from torch import nn
from torch.nn import initclass SEAttention(nn.Module):def __init__(self, channel=512