提交 b33f4c82 编写于 作者: Z Zhi Tian

rename to fcos_core and use FCOS_syncbn_bs32_c128_MNV2_FPN_1x in demo by default

上级 e84b2c19
graft maskrcnn_benchmark/csrc
graft fcos/csrc
graft fcos/configs
......@@ -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
)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册