model_inception_v3: Inception v3 model

View source: R/models-inception.R

model_inception_v3R Documentation

Inception v3 model

Description

Architecture from Rethinking the Inception Architecture for Computer Vision The required minimum input size of the model is 75x75.

Usage

model_inception_v3(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained

(bool): If True, returns a model pre-trained on ImageNet

progress

(bool): If True, displays a progress bar of the download to stderr

...

Used to pass keyword arguments to the Inception module:

  • aux_logits (bool): If True, add an auxiliary branch that can improve training. Default: True

  • transform_input (bool): If True, preprocesses the input according to the method with which it was trained on ImageNet. Default: False

Note

Important: In contrast to the other models the inception_v3 expects tensors with a size of N x 3 x 299 x 299, so ensure your images are sized accordingly.


torchvision documentation built on June 22, 2024, 11:25 a.m.