mjdietzx

import torch
from torch.autograd import Variable
import torch.nn as nn


class Bottleneck(nn.Module):
    cardinality = 32  # the size of the set of transformations

    def __init__(self, nb_channels_in, nb_channels, nb_channels_out, stride=1):
        super().__init__()

ernestum

import numpy as np
from scipy.ndimage.interpolation import map_coordinates
from scipy.ndimage.filters import gaussian_filter

def elastic_transform(image, alpha, sigma, random_state=None):
    """Elastic deformation of images as described in [Simard2003]_.
    .. [Simard2003] Simard, Steinkraus and Platt, "Best Practices for
       Convolutional Neural Networks applied to Visual Document Analysis", in
       Proc. of the International Conference on Document Analysis and
       Recognition, 2003.