MCPcopy Create free account
hub / github.com/NVlabs/ACID / __init__

Method __init__

ACID/src/encoder/unet.py:140–211  ·  view source on GitHub ↗

Arguments: in_channels: int, number of channels in the input tensor. Default is 3 for RGB images. depth: int, number of MaxPools in the U-Net. start_filts: int, number of convolutional filters for the first conv.

(self, num_classes, in_channels=3, depth=5, 
                 start_filts=64, up_mode='transpose', 
                 merge_mode='concat', **kwargs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

__init__Method · 0.45
__init__Method · 0.45

Calls 4

reset_paramsMethod · 0.95
DownConvClass · 0.85
UpConvClass · 0.85
conv1x1Function · 0.85

Tested by

no test coverage detected