Commit 1f6d5bde authored by Xiaoliang Dai's avatar Xiaoliang Dai Committed by Facebook GitHub Bot
Browse files

Add a mask and keypoint head to FBNetV3_G

Summary: Add a mask and keypoint head to FBNetV3_G to support more use cases

Reviewed By: zhanghang1989

Differential Revision: D30205169

fbshipit-source-id: 2fb88df3770fa749331021756ef5a5b6c9493bf5
parent 27f2b21d
......@@ -480,6 +480,8 @@ MODEL_ARCH_BUILTIN = {
"trunk": FBNetV3_G[0:4],
"rpn": [[_repeat_last(FBNetV3_G[3])]],
"bbox": [FBNetV3_G[4]],
"mask": LARGE_UPSAMPLE_HEAD_STAGES,
"kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_H": {
......
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