DiracNetV2

开源地址：

https://github.com/osmr/imgclsmob

网络挺快，但是辨识不收敛

__all__ = ['DiracNetV2', 'diracnet18v2', 'diracnet34v2']import os
import timeimport torch
import torch.nn as nn
import torch.nn.init as initclass DiracConv(nn.Module):def __init__(self,in_channels,out_channels,kernel_size,stride,padding):super(DiracConv, self).__init__()self.activ = nn.ReLU(inplace=True)self.conv = nn.Conv2d(in_channels=in_channels,out_channels=out_channels,kernel_size=kernel_size,stride=stride,padding=padding,bias=True)def forward(self, x):x = self.activ(x)x = self.conv(x)return xdef dir

DiracNetV2

DiracNetV2相关推荐

最新文章

热门文章