Method __init__

core/update.py:17–21

(self, input_dim=128, hidden_dim=256)

class FlowHead_cnn(nn.Module):
    def __init__(self, input_dim=128, hidden_dim=256):
        super(FlowHead_cnn, self).__init__()
        self.conv1 = nn.Conv2d(input_dim, hidden_dim, 3, padding=1)
        self.conv2 = nn.Conv2d(hidden_dim, 2, 3, padding=1)
        self.relu = nn.ReLU(inplace=True)

    def forward(self, x):
        return self.conv2(self.relu(self.conv1(x)))