View source on GitHub
|
Instantiates the RegNetY040 architecture.
tf.keras.applications.regnet.RegNetY040(
model_name='regnety040',
include_top=True,
include_preprocessing=True,
weights='imagenet',
input_tensor=None,
input_shape=None,
pooling=None,
classes=1000,
classifier_activation='softmax'
)
Reference | |
|---|---|
- Designing Network Design Spaces (CVPR 2020)
For image classification use cases, see this page for detailed examples.
For transfer learning use cases, make sure to read the guide to transfer learning & fine-tuning.
The naming of models is as follows: RegNet<block_type><flops> where
block_type is one of (X, Y) and flops signifies hundred million
floating point operations. For example RegNetY064 corresponds to RegNet with
Y block and 6.4 giga flops (64 hundred million flops).
Args |
|---|
include_top
weights
None (random initialization),
"imagenet" (pre-training on ImageNet), or the path to the weights
file to be loaded. Defaults to "imagenet".
input_tensor
layers.Input())
to use as image input for the model.
input_shape
include_top is False.
It should have exactly 3 inputs channels.
pooling
include_top is False. Defaults to None.Nonemeans that the output of the model will be the 4D tensor output of the last convolutional layer.avgmeans that global average pooling will be applied to the output of the last convolutional layer, and thus the output of the model will be a 2D tensor.maxmeans that global max pooling will be applied.classesOptional number of classes to classify images into, only to be specified if include_topis True, and if noweightsargument is specified. Defaults to 1000 (number of ImageNet classes).classifier_activationA stror callable. The activation function to use on the "top" layer. Ignored unlessinclude_top=True. Setclassifier_activation=Noneto return the logits of the "top" layer. Defaults to"softmax". When loading pretrained weights,classifier_activationcan only beNoneor"softmax".
Returns | |
|---|---|
A keras.Model instance.
|
View source on GitHub