get_conv_outsize_2d()

get_conv_outsize_2d.py

# -*- coding: utf-8 -*-

from chainer.links.connection.convolution_2d import _pair
from chainer.utils.conv import get_conv_outsize

def get_conv_outsize_2d(size, k, s, p=0, cover_all=False, d=1):
    sizeh, sizew = _pair(size)
    kh, kw = _pair(k)
    sh, sw = _pair(s)
    ph, pw = _pair(p)
    h = get_conv_outsize(sizeh, kh, sh, ph, cover_all, d)
    w = get_conv_outsize(sizew, kw, sw, pw, cover_all, d)
    return h, w