Installation

pip install GaborNet

Getting started

importtorchimporttorch.nnasnnfromtorch.nnimportfunctionalasFfromGaborNetimportGaborConv2ddevice=torch.device("cuda:0"iftorch.cuda.is_available()else"cpu")classGaborNN(nn.Module):def__init__(self):super(GaborNN,self).__init__()self.g0=GaborConv2d(in_channels=1,out_channels=96,kernel_size=(11,11))self.c1=nn.Conv2d(96,384, (3,3))self.fc1=nn.Linear(384*3*3,64)self.fc2=nn.Linear(64,2)defforward(self,x):x=F.leaky_relu(self.g0(x))x=nn.MaxPool2d()(x)x=F.leaky_relu(self.c1(x))x=nn.MaxPool2d()(x)x=x.view(-1,384*3*3)x=F.leaky_relu(self.fc1(x))x=self.fc2(x)returnxnet=GaborNN().to(device)

Citation

@misc{gabornet,    author = {Alekseev, Andrey},    title = {GaborNet: Gabor filters with learnable parameters in deep convolutional               neural networks},    year = {2019},    publisher = {GitHub},    journal = {GitHub repository},    howpublished = {\url{https://github.com/iKintosh/GaborNet}},}