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

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

import torch
import torch.nn as nn


class SimAM(torch.nn.Module):
    def __init__(self, e_lambda=1e-4):
        super

你可能感兴趣的:(算法实战,deeep,learning,YOLO,深度学习,人工智能)