目录

mmdet3d.models.utils的clip_sigmoid理解

目录

mmdet3d.models.utils的clip_sigmoid理解

Sigmoid 函数

标准的 sigmoid 函数定义为:

https://i-blog.csdnimg.cn/direct/23bf14385eaa40309b912c896e8a7039.png 容易得出结论: 取值范围(0, 1)

clip_sigmoid 是在标准的 sigmoid 函数基础上进行 裁剪(clip),即对 sigmoid 输出的结果加以限制,避免其超出特定范围。

import torch
import torch.nn.functional as F

def clip_sigmoid(x, min_val=0.01, max_val=0.99):
    return torch.clamp(F.sigmoid(x), min=min_val, max=max_val)