Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "a46ee4f31b0334fa8bc9a0385aa89fb67d3dcfb4"
Unverified Commit b547ab67 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

segmentation head: `num_filters=128` and `num_convs=0`

parent 0762a9ff
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
View file @ b547ab67
...@@ -182,6 +182,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig: ...@@ -182,6 +182,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
num_classes=num_semantic_segmentation_classes, num_classes=num_semantic_segmentation_classes,
head=SEGMENTATION_HEAD( head=SEGMENTATION_HEAD(
level=2, level=2,
num_convs=0,
num_filters=128,
decoder_min_level=2, decoder_min_level=2,
decoder_max_level=6, decoder_max_level=6,
feature_fusion='panoptic_fpn_fusion'))), feature_fusion='panoptic_fpn_fusion'))),
...@@ -199,6 +201,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig: ...@@ -199,6 +201,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
drop_remainder=False), drop_remainder=False),
annotation_file=os.path.join(_COCO_INPUT_PATH_BASE, annotation_file=os.path.join(_COCO_INPUT_PATH_BASE,
'instances_val2017.json'), 'instances_val2017.json'),
segmentation_evaluation=semantic_segmentation.Evaluation(
report_per_class_iou=False, report_train_mean_iou=False),
panoptic_quality_evaluator=PanopticQualityEvaluator( panoptic_quality_evaluator=PanopticQualityEvaluator(
num_categories=num_panoptic_categories, num_categories=num_panoptic_categories,
ignored_label=0, ignored_label=0,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment