mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
3a785f1223
* [WIP] Refactor v2.0 (#163) * Refactor backend wrapper * Refactor mmdet.inference * Fix * merge * refactor utils * Use deployer and deploy_model to manage pipeline * Resolve comments * Add a real inference api function * rename wrappers * Set execute to private method * Rename deployer deploy_model * Refactor task * remove type hint * lint * Resolve comments * resolve comments * lint * docstring * [Fix]: Fix bugs in details in refactor branch (#192) * [WIP] Refactor v2.0 (#163) * Refactor backend wrapper * Refactor mmdet.inference * Fix * merge * refactor utils * Use deployer and deploy_model to manage pipeline * Resolve comments * Add a real inference api function * rename wrappers * Set execute to private method * Rename deployer deploy_model * Refactor task * remove type hint * lint * Resolve comments * resolve comments * lint * docstring * Fix errors * lint * resolve comments * fix bugs * conflict * lint and typo * Resolve comment * refactor mmseg (#201) * support mmseg * fix docstring * fix docstring * [Refactor]: Get the count of backend files (#202) * Fix backend files * resolve comments * lint * Fix ncnn * [Refactor]: Refactor folders of mmdet (#200) * Move folders * lint * test object detection model * lint * reset changes * fix openvino * resolve comments * __init__.py * Fix path * [Refactor]: move mmseg (#206) * [Refactor]: Refactor mmedit (#205) * feature mmedit * edit2.0 * edit * refactor mmedit * fix __init__.py * fix __init__ * fix formai * fix comment * fix comment * Fix wrong func_name of ConvFCBBoxHead (#209) * [Refactor]: Refactor mmdet unit test (#207) * Move folders * lint * test object detection model * lint * WIP * remove print * finish unit test * Fix tests * resolve comments * Add mask test * lint * resolve comments * Refine cfg file * Move files * add files * Fix path * [Unittest]: Refine the unit tests in mmdet #214 * [Refactor] refactor mmocr to mmdeploy/codebase (#213) * refactor mmocr to mmdeploy/codebase * fix docstring of show_result * fix docstring of visualize * refine docstring * replace print with logging * refince codes * resolve comments * resolve comments * [Refactor]: mmseg tests (#210) * refactor mmseg tests * rename test_codebase * update * add model.py * fix * [Refactor] Refactor mmcls and the package (#217) * refactor mmcls * fix yapf * fix isort * refactor-mmcls-package * fix print to logging * fix docstrings according to others comments * fix comments * fix comments * fix allentdans comment in pr215 * remove mmocr init * [Refactor] Refactor mmedit tests (#212) * feature mmedit * edit2.0 * edit * refactor mmedit * fix __init__.py * fix __init__ * fix formai * fix comment * fix comment * buff * edit test and code refactor * refactor dir * refactor tests/mmedit * fix docstring * add test coverage * fix lint * fix comment * fix comment * Update typehint (#216) * update type hint * update docstring * update * remove file * fix ppl * Refine get_predefined_partition_cfg * fix tensorrt version > 8 * move parse_cuda_device_id to device.py * Fix cascade * onnx2ncnn docstring Co-authored-by: Yifan Zhou <singlezombie@163.com> Co-authored-by: RunningLeon <maningsheng@sensetime.com> Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com> Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com> Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
232 lines
5.7 KiB
JSON
232 lines
5.7 KiB
JSON
{
|
|
"type": "MaskRCNN",
|
|
"backbone": {
|
|
"type": "ResNet",
|
|
"depth": 50,
|
|
"num_stages": 4,
|
|
"out_indices": [
|
|
0,
|
|
1,
|
|
2,
|
|
3
|
|
],
|
|
"frozen_stages": 1,
|
|
"norm_cfg": {
|
|
"type": "BN",
|
|
"requires_grad": true
|
|
},
|
|
"norm_eval": true,
|
|
"style": "pytorch",
|
|
"init_cfg": {
|
|
"type": "Pretrained",
|
|
"checkpoint": "torchvision://resnet50"
|
|
}
|
|
},
|
|
"neck": {
|
|
"type": "FPN",
|
|
"in_channels": [
|
|
256,
|
|
512,
|
|
1024,
|
|
2048
|
|
],
|
|
"out_channels": 256,
|
|
"num_outs": 5
|
|
},
|
|
"rpn_head": {
|
|
"type": "RPNHead",
|
|
"in_channels": 256,
|
|
"feat_channels": 256,
|
|
"anchor_generator": {
|
|
"type": "AnchorGenerator",
|
|
"scales": [
|
|
8
|
|
],
|
|
"ratios": [
|
|
0.5,
|
|
1.0,
|
|
2.0
|
|
],
|
|
"strides": [
|
|
4,
|
|
8,
|
|
16,
|
|
32,
|
|
64
|
|
]
|
|
},
|
|
"bbox_coder": {
|
|
"type": "DeltaXYWHBBoxCoder",
|
|
"target_means": [
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0
|
|
],
|
|
"target_stds": [
|
|
1.0,
|
|
1.0,
|
|
1.0,
|
|
1.0
|
|
]
|
|
},
|
|
"loss_cls": {
|
|
"type": "CrossEntropyLoss",
|
|
"use_sigmoid": true,
|
|
"loss_weight": 1.0
|
|
},
|
|
"loss_bbox": {
|
|
"type": "L1Loss",
|
|
"loss_weight": 1.0
|
|
}
|
|
},
|
|
"roi_head": {
|
|
"type": "StandardRoIHead",
|
|
"bbox_roi_extractor": {
|
|
"type": "SingleRoIExtractor",
|
|
"roi_layer": {
|
|
"type": "RoIAlign",
|
|
"output_size": 7,
|
|
"sampling_ratio": 0
|
|
},
|
|
"out_channels": 256,
|
|
"featmap_strides": [
|
|
4,
|
|
8,
|
|
16,
|
|
32
|
|
]
|
|
},
|
|
"bbox_head": {
|
|
"type": "Shared2FCBBoxHead",
|
|
"in_channels": 256,
|
|
"fc_out_channels": 1024,
|
|
"roi_feat_size": 7,
|
|
"num_classes": 80,
|
|
"bbox_coder": {
|
|
"type": "DeltaXYWHBBoxCoder",
|
|
"target_means": [
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0
|
|
],
|
|
"target_stds": [
|
|
0.1,
|
|
0.1,
|
|
0.2,
|
|
0.2
|
|
]
|
|
},
|
|
"reg_class_agnostic": false,
|
|
"loss_cls": {
|
|
"type": "CrossEntropyLoss",
|
|
"use_sigmoid": false,
|
|
"loss_weight": 1.0
|
|
},
|
|
"loss_bbox": {
|
|
"type": "L1Loss",
|
|
"loss_weight": 1.0
|
|
}
|
|
},
|
|
"mask_roi_extractor": {
|
|
"type": "SingleRoIExtractor",
|
|
"roi_layer": {
|
|
"type": "RoIAlign",
|
|
"output_size": 14,
|
|
"sampling_ratio": 0
|
|
},
|
|
"out_channels": 256,
|
|
"featmap_strides": [
|
|
4,
|
|
8,
|
|
16,
|
|
32
|
|
]
|
|
},
|
|
"mask_head": {
|
|
"type": "FCNMaskHead",
|
|
"num_convs": 4,
|
|
"in_channels": 256,
|
|
"conv_out_channels": 256,
|
|
"num_classes": 80,
|
|
"loss_mask": {
|
|
"type": "CrossEntropyLoss",
|
|
"use_mask": true,
|
|
"loss_weight": 1.0
|
|
}
|
|
}
|
|
},
|
|
"train_cfg": {
|
|
"rpn": {
|
|
"assigner": {
|
|
"type": "MaxIoUAssigner",
|
|
"pos_iou_thr": 0.7,
|
|
"neg_iou_thr": 0.3,
|
|
"min_pos_iou": 0.3,
|
|
"match_low_quality": true,
|
|
"ignore_iof_thr": -1
|
|
},
|
|
"sampler": {
|
|
"type": "RandomSampler",
|
|
"num": 256,
|
|
"pos_fraction": 0.5,
|
|
"neg_pos_ub": -1,
|
|
"add_gt_as_proposals": false
|
|
},
|
|
"allowed_border": -1,
|
|
"pos_weight": -1,
|
|
"debug": false
|
|
},
|
|
"rpn_proposal": {
|
|
"nms_pre": 2000,
|
|
"max_per_img": 1000,
|
|
"nms": {
|
|
"type": "nms",
|
|
"iou_threshold": 0.7
|
|
},
|
|
"min_bbox_size": 0
|
|
},
|
|
"rcnn": {
|
|
"assigner": {
|
|
"type": "MaxIoUAssigner",
|
|
"pos_iou_thr": 0.5,
|
|
"neg_iou_thr": 0.5,
|
|
"min_pos_iou": 0.5,
|
|
"match_low_quality": true,
|
|
"ignore_iof_thr": -1
|
|
},
|
|
"sampler": {
|
|
"type": "RandomSampler",
|
|
"num": 512,
|
|
"pos_fraction": 0.25,
|
|
"neg_pos_ub": -1,
|
|
"add_gt_as_proposals": true
|
|
},
|
|
"mask_size": 28,
|
|
"pos_weight": -1,
|
|
"debug": false
|
|
}
|
|
},
|
|
"test_cfg": {
|
|
"rpn": {
|
|
"nms_pre": 1000,
|
|
"max_per_img": 1000,
|
|
"nms": {
|
|
"type": "nms",
|
|
"iou_threshold": 0.7
|
|
},
|
|
"min_bbox_size": 0
|
|
},
|
|
"rcnn": {
|
|
"score_thr": 0.05,
|
|
"nms": {
|
|
"type": "nms",
|
|
"iou_threshold": 0.5
|
|
},
|
|
"max_per_img": 100,
|
|
"mask_thr_binary": 0.5
|
|
}
|
|
}
|
|
}
|