private
vgg_classifier_layers
— function
vgg_classifier_layers(imsize, nclasses, fcsize, dropout)
Create VGG classifier (fully connected) layers (reference).
Arguments
imsize
: tuple(width, height, channels)
indicating the size after the convolution layers (seeMetalhead.vgg_convolutional_layers
)nclasses
: number of output classesfcsize
: input and output size of the intermediate fully connected layerdropout
: the dropout level between each fully connected layer