public
VGG
— struct
VGG(imsize = (224, 224); config, inchannels, batchnorm = false, nclasses, fcsize, dropout)
Create a VGG
model
(reference).
See also vgg
.
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