Skip to content

Commit d228af5

Browse files
committed
Add unified header to roi align rotated
1 parent a15f377 commit d228af5

File tree

4 files changed

+8
-5
lines changed

4 files changed

+8
-5
lines changed

mmcv/ops/csrc/onnxruntime/cpu/roi_align_rotated.cpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1+
// Modified from
2+
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
3+
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
14
#include "roi_align_rotated.h"
2-
35
#include "../ort_mmcv_utils.h"
46

5-
// Implementation taken from
6-
// https://github.com/facebookresearch/detectron2/blob/master/detectron2/layers/csrc/ROIAlignRotated/ROIAlignRotated_cpu.cpp
77
struct PreCalc {
88
int pos1;
99
int pos2;

mmcv/ops/csrc/parrots/roi_align_rotated_cpu.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
// Modified from
2-
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlign
2+
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
33
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
44
#include <ATen/ATen.h>
55
#include <ATen/TensorUtils.h>

mmcv/ops/csrc/pytorch/roi_align_rotated_cpu.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
// Modified from
2-
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlign
2+
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
33
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
44
#include <ATen/ATen.h>
55
#include <ATen/TensorUtils.h>

mmcv/ops/csrc/roi_align_rotated_cuda_kernel.cuh

+3
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
// Modified from
2+
// https://github.com/facebookresearch/detectron2/tree/master/detectron2/layers/csrc/ROIAlignRotated
3+
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
14
#ifndef ROI_ALIGN_ROTATED_CUDA_KERNEL_CUH
25
#define ROI_ALIGN_ROTATED_CUDA_KERNEL_CUH
36

0 commit comments

Comments
 (0)