From b8253153c233348ee4bc0714252a8b0e5e4e7bcc Mon Sep 17 00:00:00 2001 From: whytin Date: Thu, 11 Jul 2019 11:22:34 +0800 Subject: [PATCH] Update model_builder.py #L279 Correct RoIAlignFunction() to ROIAlgin() --- lib/modeling/model_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/modeling/model_builder.py b/lib/modeling/model_builder.py index f1f5c02..e8c1279 100644 --- a/lib/modeling/model_builder.py +++ b/lib/modeling/model_builder.py @@ -276,7 +276,7 @@ def roi_feature_transform(self, blobs_in, rpn_ret, blob_rois='rois', method='RoI # Warning!: Not check if implementation matches Detectron xform_out = ROIPool((resolution, resolution), sc)(bl_in, rois) elif method == 'RoIAlign': - xform_out = RoIAlignFunction( + xform_out = ROIAlign( (resolution, resolution), sc, sampling_ratio)(bl_in, rois) bl_out_list.append(xform_out)