Struct std::UNet

#include <std/ml/unet.h> source

struct UNet {
    struct Conv2D encConv1;
    struct Conv2D encConv2;
    struct Conv2D bottleneck;
    struct Conv2D decConv1;
    struct Conv2D decConv2;
}

Fields

§struct Conv2D encConv1source

inChannels -> c1

§struct Conv2D encConv2source

c1 -> c2

§struct Conv2D bottlenecksource

c2 -> c2

§struct Conv2D decConv1source

(c2 + c2 skip) -> c1

§struct Conv2D decConv2source

(c1 + c1 skip) -> outChannels

Methods

§struct FeatureMap forward(const &FeatureMap input)source · definition

Input height/width must each be divisible by 4 (two 2x poolings). 'net' isn't semantically mutated, but is taken by mutable reference since its Conv2D sub-layers are threaded through by address internally.

§void drop()source · definition

§void free() consumesource · definition

= drop()

§

static struct UNet new(
    unsigned long inChannels,
    unsigned long c1,
    unsigned long c2,
    unsigned long outChannels,
)

source · definition