开源地址:
https://github.com/osmr/imgclsmob
网络挺快,但是辨识不收敛
__all__ = ['DiracNetV2', 'diracnet18v2', 'diracnet34v2']
import os
import time
import torch
import torch.nn as nn
import torch.nn.init as init
class 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,