RefineDet
一昂Yang
这个作者很懒,什么都没留下…
展开
-
torch.autograd.Function
pytorch自定义layer有两种方式: 方式一:通过继承torch.nn.Module类来实现拓展。只需重新实现__init__和forward函数。 方式二:通过继承torch.autograd.Function,除了要实现__init__和forward函数,还要实现backward函数(就是自定义求导规则)。 方式一看着更简单一点,torch.nn.Module不香么?为毛要用方式...原创 2020-01-09 15:44:17 · 3496 阅读 · 0 评论 -
pytorch-RefineDet------PriorBox
相关函数 1、torch.clamp() torch.clamp(input, min, max, out=None) --> Tensor 将输入input Tensor中每个元素约束到【min,max】,返回一个截断后的new Tensor 例子: input = torch.Tensor([2,1,0.5,-0.5]) output = torch.clamp(input...原创 2020-01-01 17:25:30 · 368 阅读 · 0 评论