efficientdet_model {fastai}

Eficientdet model

Description

Creates the efficientdet model specified by 'model_name'.

Usage

efficientdet_model(model_name, num_classes, img_size, pretrained = TRUE)

Arguments

model_name

Specifies the model to create. For pretrained models, check [this](https://github.com/rwightman/efficientdet-pytorch#models) table.

num_classes

Number of classes of your dataset (including background).

img_size

Image size that will be fed to the model. Must be squared and divisible by 128.

pretrained

If TRUE, use a pretrained backbone (on COCO).

Value

model