imgcook / pipcook-plugin-detectron-fasterrcnn-model

Pipcook plugin for fasterrcnn detectron model definition.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

@pipcook/plugins-detectron-fasterrcnn-model-define

Description

This plugin defines object detection model. This model is fasterrcnn. The whole staff is built on top on detectron2.

Required Parameters

Parameter Type Description

Optional Parameters

Parameter Type Default Value Description
baseLearningRate number 0.00025 base learning rate
numWorkers number 0 workers for data loader.
numClasses number 0 Number of classes of dataset. Only specify this value when you provide your own model. Otherwise just leave as 0
recoverPath string null path to model pretrained

Example

"modelDefine": {
  "package": "@pipcook/plugins-detectron-fasterrcnn-model-define"
},

About

Pipcook plugin for fasterrcnn detectron model definition.

License:Apache License 2.0


Languages

Language:TypeScript 100.0%