Commit ba7c235b authored by Debojeet Chatterjee's avatar Debojeet Chatterjee Committed by Facebook GitHub Bot
Browse files

Add new FBNetV3_B_large backbone for HAPTIC.

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/659

FBNetV3_B_large using FBNetV3_B backbone with large box and kpts heads for better multitask KP+Box detection tasks while being in similar range of #flops and #params.

Reviewed By: ashishvshenoy

Differential Revision: D55645349

fbshipit-source-id: 3fe84f566b3eeaddf84a94ef708557944fffcd22
parent dda734a4
...@@ -546,6 +546,14 @@ MODEL_ARCH_BUILTIN = { ...@@ -546,6 +546,14 @@ MODEL_ARCH_BUILTIN = {
"kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES, "kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES,
"basic_args": _BASIC_ARGS, "basic_args": _BASIC_ARGS,
}, },
"FBNetV3_B_large": {
"trunk": FBNetV3_B_no_se[0:4],
"rpn": [[_repeat_last(FBNetV3_B_no_se[3])]],
"bbox": LARGE_BOX_HEAD_STAGES,
"mask": SMALL_DS_UPSAMPLE_HEAD_STAGES,
"kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_B_light_no_se_C5": { "FBNetV3_B_light_no_se_C5": {
"trunk": FBNetV3_B_light_no_se[0:5], "trunk": FBNetV3_B_light_no_se[0:5],
"rpn": [[_repeat_last(FBNetV3_B_light_no_se[3])]], "rpn": [[_repeat_last(FBNetV3_B_light_no_se[3])]],
......
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