diff --git a/MANIFEST.in b/MANIFEST.in index 6bc166a65637cfe77a5c957458154e215ccb7441..ba74cb5a2068365f9ac16946213abe319ac9f642 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -1,2 +1,2 @@ -graft maskrcnn_benchmark/csrc +graft fcos/csrc graft fcos/configs diff --git a/fcos/bin/fcos b/fcos/bin/fcos index 042fc0cc8903930f4f824039c0bd6a87c68ddb61..e2d309ab6cc7e51f157e4e81447ac287d922b8c7 100644 --- 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 )