efficientdet_model: Eficientdet model

Description Usage Arguments Value

View source: R/icevision_models.R

Description

Creates the efficientdet model specified by 'model_name'.

Usage

1
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


fastai documentation built on Oct. 25, 2021, 5:08 p.m.