CSPDarKNet代码实现(pytorch)
CSPDarkNet53代码实现(pytorch)
import torch.nn as nn
import torch
import torch.nn.functional as F
// 激活函数Mish
class Mish(nn.Module):
def __init__(self):
super(Mish, self).__init__()
def forward(self, x):
return x * torch.tanh(F.softp
原创
2021-03-18 20:39:22 ·
1354 阅读 ·
0 评论