rename to fcos_core and use FCOS_syncbn_bs32_c128_MNV2_FPN_1x in demo by default

b33f4c82 · Zhi Tian · e84b2c19 · b33f4c82 · b33f4c82
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

MANIFEST.in MANIFEST.in +1 -1

fcos/bin/fcos fcos/bin/fcos +1 -1

未找到文件。
--- a/MANIFEST.in
+++ b/MANIFEST.in
-graft maskrcnn_benchmark/csrc
+graft fcos/csrc
 graft fcos/configs
--- a/fcos/bin/fcos
+++ b/fcos/bin/fcos
@@ -31,7 +31,7 @@ parser.add_argument(
 args = parser.parse_args()
 fcos = FCOS(
-    model_name="fcos_R_50_FPN_1x",
+    model_name="fcos_syncbn_bs32_c128_MNV2_FPN_1x",
    nms_thresh=0.6,
    cpu_only=not torch.cuda.is_available()  # if you do not have GPUs, please set cpu_only as True
 )