yuhsuanyeh / BiFuse

[CVPR2020] BiFuse: Monocular 360 Depth Estimation via Bi-Projection Fusion
MIT License
173 stars 28 forks source link

Fix access _output_size_init via self in FCRN.py #1

Open l3str4nge opened 4 years ago

l3str4nge commented 4 years ago

Hello

In models/FCRN.py:

class fusion_ResNet(nn.Module):
    _output_size_init = (256, 256)

    def __init__(self, bs, layers, decoder, output_size=None, in_channels=3, pretrained=True, padding='ZeroPad'):

        ...
        if output_size == None:
            output_size = _output_size_init  # undefined variable, changed to self._output_size_init