View Source ExVision.InstanceSegmentation.MaskRCNN_ResNet50_FPN_V2 (Ex Vision v0.4.0)

An instance segmentation model with a ResNet-50-FPN backbone. Exported from torchvision.

Summary

Types

Type describing all categories recognised by MaskRCNN ResNet50 FPN V2

t()

An instance of the Elixir.ExVision.InstanceSegmentation.MaskRCNN_ResNet50_FPN_V2

Functions

Submits the input for inference to the process running the Nx.Serving for this model.

Returns a list of all categories recognised by MaskRCNN ResNet50 FPN V2

Creates the model instance

Same as load/1, but raises and error on failure.

Immediatelly applies the model to the given input, in the scope of the current process.

Types

@type category_t() ::
  :__background__
  | :person
  | :bicycle
  | :car
  | :motorcycle
  | :airplane
  | :bus
  | :train
  | :truck
  | :boat
  | :traffic_light
  | :fire_hydrant
  | :"n/a"
  | :stop_sign
  | :parking_meter
  | :bench
  | :bird
  | :cat
  | :dog
  | :horse
  | :sheep
  | :cow
  | :elephant
  | :bear
  | :zebra
  | :giraffe
  | :backpack
  | :umbrella
  | :handbag
  | :tie
  | :suitcase
  | :frisbee
  | :skis
  | :snowboard
  | :sports_ball
  | :kite
  | :baseball_bat
  | :baseball_glove
  | :skateboard
  | :surfboard
  | :tennis_racket
  | :bottle
  | :wine_glass
  | :cup
  | :fork
  | :knife
  | :spoon
  | :bowl
  | :banana
  | :apple
  | :sandwich
  | :orange
  | :broccoli
  | :carrot
  | :hot_dog
  | :pizza
  | :donut
  | :cake
  | :chair
  | :couch
  | :potted_plant
  | :bed
  | :dining_table
  | :toilet
  | :tv
  | :laptop
  | :mouse
  | :remote
  | :keyboard
  | :cell_phone
  | :microwave
  | :oven
  | :toaster
  | :sink
  | :refrigerator
  | :book
  | :clock
  | :vase
  | :scissors
  | :teddy_bear
  | :hair_drier
  | :toothbrush

Type describing all categories recognised by MaskRCNN ResNet50 FPN V2

@type output_t() :: [ExVision.Types.BBoxWithMask.t()]
@opaque t()

An instance of the Elixir.ExVision.InstanceSegmentation.MaskRCNN_ResNet50_FPN_V2

Functions

Link to this function

batched_run(name \\ __MODULE__, input)

View Source
@spec batched_run(atom(), ExVision.Model.input_t()) :: output_t() | [output_t()]

Submits the input for inference to the process running the Nx.Serving for this model.

@spec categories() :: [category_t()]

Returns a list of all categories recognised by MaskRCNN ResNet50 FPN V2

@spec load([ExVision.Model.Definition.Ortex.load_option_t()]) ::
  {:ok, t()} | {:error, reason :: atom()}

Creates the model instance

@spec load!(keyword()) :: t()

Same as load/1, but raises and error on failure.

@spec run(t(), ExVision.Model.input_t()) :: output_t() | [output_t()]

Immediatelly applies the model to the given input, in the scope of the current process.