diff --git a/src/operator/contrib/mrcnn_mask_target.cu b/src/operator/contrib/mrcnn_mask_target.cu index e85b6a5e2c95..d913f2486d44 100644 --- a/src/operator/contrib/mrcnn_mask_target.cu +++ b/src/operator/contrib/mrcnn_mask_target.cu @@ -146,7 +146,7 @@ __device__ void RoIAlignForward( T roi_height = roi_end_h - roi_start_h; // Force malformed ROIs to be 1x1 - if (!continuous_coordinate) { // backward compatiblity + if (!continuous_coordinate) { // backward compatiblity // Force malformed ROIs to be 1x1 roi_width = max(roi_width, (T)1.); roi_height = max(roi_height, (T)1.); @@ -260,7 +260,7 @@ void MRCNNMaskTargetRun(const MRCNNMaskTargetParam& param, const std::vecto MRCNNMaskTargetKernel<<>> (rois.dptr_, gt_masks.dptr_, matches.dptr_, cls_targets.dptr_, - out_masks.dptr_, out_mask_cls.dptr_, num_el, param.aligned, + out_masks.dptr_, out_mask_cls.dptr_, num_el, param.aligned, batch_size, param.num_classes, param.num_rois, num_gtmasks, gt_height, gt_width, param.mask_size[0], param.mask_size[1], param.sample_ratio); diff --git a/src/operator/contrib/roi_align.cc b/src/operator/contrib/roi_align.cc index 4741be189efc..eeb9f622c2be 100644 --- a/src/operator/contrib/roi_align.cc +++ b/src/operator/contrib/roi_align.cc @@ -187,7 +187,7 @@ num_threads(engine::OpenMP::Get()->GetRecommendedOMPThreadCount()) if (continuous_coordinate) { CHECK_GT(roi_width, 0.); CHECK_GT(roi_height, 0.); - } else { // backward compatiblity + } else { // backward compatiblity // Force malformed ROIs to be 1x1 roi_width = std::max(roi_width, (T)1.); roi_height = std::max(roi_height, (T)1.); @@ -367,7 +367,7 @@ void ROIAlignBackward( T roi_width = roi_end_w - roi_start_w; T roi_height = roi_end_h - roi_start_h; - if (!continuous_coordinate) { // backward compatiblity + if (!continuous_coordinate) { // backward compatiblity // Force malformed ROIs to be 1x1 roi_width = std::max(roi_width, (T)1.); roi_height = std::max(roi_height, (T)1.); diff --git a/src/operator/contrib/roi_align.cu b/src/operator/contrib/roi_align.cu index 8cf5f1f5efe9..3d085d3031b5 100644 --- a/src/operator/contrib/roi_align.cu +++ b/src/operator/contrib/roi_align.cu @@ -142,7 +142,7 @@ __global__ void RoIAlignForwardKernel( T roi_width = roi_end_w - roi_start_w; T roi_height = roi_end_h - roi_start_h; - if (!continuous_coordinate) { // backward compatiblity + if (!continuous_coordinate) { // backward compatiblity // Force malformed ROIs to be 1x1 roi_width = max(roi_width, (T)1.); roi_height = max(roi_height, (T)1.); @@ -284,7 +284,7 @@ __global__ void RoIAlignBackwardKernel( T roi_width = roi_end_w - roi_start_w; T roi_height = roi_end_h - roi_start_h; - if (!continuous_coordinate) { // backward compatiblity + if (!continuous_coordinate) { // backward compatiblity // Force malformed ROIs to be 1x1 roi_width = max(roi_width, (T)1.); roi_height = max(roi_height, (T)1.);