@pipcook/plugins-detectron-maskrcnn-model-define
Description
This plugin defines object detection / instance segmentation model. This model is maskrcnn. 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-maskrcnn-model-define"
},