Add unified header to roi align rotated

open-mmlab · Apr 19, 2021 · d228af5 · d228af5
1 parent a15f377
commit d228af5
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/mmcv/ops/csrc/onnxruntime/cpu/roi_align_rotated.cpp b/mmcv/ops/csrc/onnxruntime/cpu/roi_align_rotated.cpp
@@ -1,9 +1,9 @@
+// Modified from
+// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
+// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 #include "roi_align_rotated.h"
-
 #include "../ort_mmcv_utils.h"
 
-// Implementation taken from
-// https://github.com/facebookresearch/detectron2/blob/master/detectron2/layers/csrc/ROIAlignRotated/ROIAlignRotated_cpu.cpp
 struct PreCalc {
   int pos1;
   int pos2;

diff --git a/mmcv/ops/csrc/parrots/roi_align_rotated_cpu.cpp b/mmcv/ops/csrc/parrots/roi_align_rotated_cpu.cpp
@@ -1,5 +1,5 @@
 // Modified from
-// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlign
+// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
 // Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 #include <ATen/ATen.h>
 #include <ATen/TensorUtils.h>

diff --git a/mmcv/ops/csrc/pytorch/roi_align_rotated_cpu.cpp b/mmcv/ops/csrc/pytorch/roi_align_rotated_cpu.cpp
@@ -1,5 +1,5 @@
 // Modified from
-// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlign
+// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
 // Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 #include <ATen/ATen.h>
 #include <ATen/TensorUtils.h>

diff --git a/mmcv/ops/csrc/roi_align_rotated_cuda_kernel.cuh b/mmcv/ops/csrc/roi_align_rotated_cuda_kernel.cuh
@@ -1,3 +1,6 @@
+// Modified from
+// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
+// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
 #ifndef ROI_ALIGN_ROTATED_CUDA_KERNEL_CUH
 #define ROI_ALIGN_ROTATED_CUDA_KERNEL_CUH