[Feature] Add 6 rotated detection ops in parrots (#1665)

* add active_rotated_filter, convex_iou, min_area_polygons, points_in_polygons, riroi_align_roteted, rotated_feature_align in parrots

* fix lint

* fix lint

mmcv/ops/riroi_align_rotated.py

@@ -41,10 +41,16 @@ class RiRoIAlignRotatedFunction(Function):
 
         output = features.new_zeros(num_rois, num_channels, out_h, out_w)
 
-        ext_module.riroi_align_rotated_forward(features, rois, output, out_h,
-                                               out_w, spatial_scale,
-                                               num_samples, num_orientations,
-                                               clockwise)
+        ext_module.riroi_align_rotated_forward(
+            features,
+            rois,
+            output,
+            pooled_height=out_h,
+            pooled_width=out_w,
+            spatial_scale=spatial_scale,
+            num_samples=num_samples,
+            num_orientations=num_orientations,
+            clockwise=clockwise)
         return output
 
     @staticmethod
@@ -67,8 +73,15 @@ class RiRoIAlignRotatedFunction(Function):
             grad_input = rois.new_zeros(batch_size, num_channels, feature_h,
                                         feature_w)
             ext_module.riroi_align_rotated_backward(
-                grad_output.contiguous(), rois, grad_input, out_h, out_w,
-                spatial_scale, num_samples, num_orientations, clockwise)
+                grad_output.contiguous(),
+                rois,
+                grad_input,
+                pooled_height=out_h,
+                pooled_width=out_w,
+                spatial_scale=spatial_scale,
+                num_samples=num_samples,
+                num_orientations=num_orientations,
+                clockwise=clockwise)
 
             return grad_input, grad_rois, None, None, None, None, None
 

mmcv/ops/rotated_feature_align.py

@@ -40,8 +40,12 @@ class RotatedFeatureAlignFunction(Function):
         ctx.save_for_backward(best_rbboxes)
         assert points in [1, 5]
         output = torch.zeros_like(features)
-        ext_module.rotated_feature_align_forward(features, best_rbboxes,
-                                                 output, spatial_scale, points)
+        ext_module.rotated_feature_align_forward(
+            features,
+            best_rbboxes,
+            output,
+            spatial_scale=spatial_scale,
+            points=points)
         return output
 
     @staticmethod
@@ -61,9 +65,12 @@ class RotatedFeatureAlignFunction(Function):
         grad_input = None
         if ctx.needs_input_grad[0]:
             grad_input = torch.zeros_like(grad_output)
-            ext_module.rotated_feature_align_backward(grad_output.contiguous(),
-                                                      best_rbboxes, grad_input,
-                                                      spatial_scale, points)
+            ext_module.rotated_feature_align_backward(
+                grad_output.contiguous(),
+                best_rbboxes,
+                grad_input,
+                spatial_scale=spatial_scale,
+                points=points)
         return grad_input, None, None, None