This plugin defines object detection model. This model is fasterrcnn. The whole staff is built on top on detectron2.
| Parameter | Type | Description |
|---|
| 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 |
"modelDefine": {
"package": "@pipcook/plugins-detectron-fasterrcnn-model-define"
},