efficientdet_model: Eficientdet model

View source: R/icevision_models.R

efficientdet_modelR Documentation

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


fastai documentation built on March 21, 2022, 9:07 a.m.