private
vgg
— function
vgg(imsize; config, inchannels, batchnorm = false, nclasses, fcsize, dropout)
Create a VGG model (reference).
Arguments
imsize
: input image width and height as a tupleconfig
: the configuration for the convolution layers (seeMetalhead.vgg_convolutional_layers
)inchannels
: number of input channelsbatchnorm
: set totrue
to use batch normalization after each convolutionnclasses
: number of output classesfcsize
: intermediate fully connected layer size (seeMetalhead.vgg_classifier_layers
)dropout
: dropout level between fully connected layers