Introduction

Tutorials

Developer guide

API Reference

private vgg_classifier_layersfunction

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 (see Metalhead.vgg_convolutional_layers)
  • nclasses: number of output classes
  • fcsize: input and output size of the intermediate fully connected layer
  • dropout: the dropout level between each fully connected layer