Commit 72a3804c authored by Hongyu Fu's avatar Hongyu Fu Committed by Facebook GitHub Bot
Browse files

land the fbnet architectures for video logo detection project

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

land the fbnet architectures for video logo detection project in RCNN, FCOS and YOLO

Reviewed By: wat3rBro

Differential Revision: D38139055

fbshipit-source-id: 6ba21f482ed067c52d438e0c217e523896c2131c
parent ab50d5d9
......@@ -437,18 +437,37 @@ MODEL_ARCH_BUILTIN = {
"mask": SMALL_DS_UPSAMPLE_HEAD_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_A_C5": {
"trunk": FBNetV3_A[0:5],
"rpn": [[_repeat_last(FBNetV3_A[3])]],
"bbox": [FBNetV3_A[4]],
"mask": SMALL_DS_UPSAMPLE_HEAD_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_B": {
"trunk": FBNetV3_B[0:4],
"rpn": [[_repeat_last(FBNetV3_B[3])]],
"bbox": [FBNetV3_B[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_B_C5": {
"trunk": FBNetV3_B[0:5],
"rpn": [[_repeat_last(FBNetV3_B[3])]],
"bbox": [FBNetV3_B[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_C": {
"trunk": FBNetV3_C[0:4],
"rpn": [[_repeat_last(FBNetV3_C[3])]],
"bbox": [FBNetV3_C[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_C_C5": {
"trunk": FBNetV3_C[0:5],
"rpn": [[_repeat_last(FBNetV3_C[3])]],
"bbox": [FBNetV3_C[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_D": {
"trunk": FBNetV3_D[0:4],
"rpn": [[_repeat_last(FBNetV3_D[3])]],
......@@ -467,6 +486,12 @@ MODEL_ARCH_BUILTIN = {
"bbox": [FBNetV3_F[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_F_C5": {
"trunk": FBNetV3_F[0:5],
"rpn": [[_repeat_last(FBNetV3_F[3])]],
"bbox": [FBNetV3_F[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_G": {
"trunk": FBNetV3_G[0:4],
"rpn": [[_repeat_last(FBNetV3_G[3])]],
......@@ -475,6 +500,14 @@ MODEL_ARCH_BUILTIN = {
"kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_G_C5": {
"trunk": FBNetV3_G[0:5],
"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": {
"trunk": FBNetV3_H[0:4],
"rpn": [[_repeat_last(FBNetV3_H[3])]],
......@@ -513,6 +546,12 @@ MODEL_ARCH_BUILTIN = {
"kpts": LARGE_UPSAMPLE_HEAD_D21_STAGES,
"basic_args": _BASIC_ARGS,
},
"FBNetV3_B_light_no_se_C5": {
"trunk": FBNetV3_B_light_no_se[0:5],
"rpn": [[_repeat_last(FBNetV3_B_light_no_se[3])]],
"bbox": [FBNetV3_B_light_no_se[4]],
"basic_args": _BASIC_ARGS,
},
"FBNetV3_G_fpn": {
"trunk": FBNetV3_G[0:5], # FPN uses all 5 stages
"rpn": [[_repeat_last(FBNetV3_G[3], n=1)]],
......
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