[Docs] Add header for files (#1181)

* [Docs] Add header for files

* change to OpenMMLab

* add headers for .cpp, .cu, .h, .cuh

* replace Open-MMLab with OpenMMLab
pull/1206/head^2
Jerry Jiarui XU 4 years ago committed by GitHub
parent 04daea425b
commit c47c9196d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      LICENSE
  2. 2
      mmcv/__init__.py
  3. 2
      mmcv/arraymisc/__init__.py
  4. 2
      mmcv/arraymisc/quantization.py
  5. 2
      mmcv/cnn/__init__.py
  6. 2
      mmcv/cnn/alexnet.py
  7. 1
      mmcv/cnn/bricks/__init__.py
  8. 1
      mmcv/cnn/bricks/activation.py
  9. 1
      mmcv/cnn/bricks/context_block.py
  10. 1
      mmcv/cnn/bricks/conv.py
  11. 1
      mmcv/cnn/bricks/conv2d_adaptive_padding.py
  12. 1
      mmcv/cnn/bricks/conv_module.py
  13. 1
      mmcv/cnn/bricks/conv_ws.py
  14. 1
      mmcv/cnn/bricks/depthwise_separable_conv_module.py
  15. 1
      mmcv/cnn/bricks/drop.py
  16. 1
      mmcv/cnn/bricks/generalized_attention.py
  17. 1
      mmcv/cnn/bricks/hsigmoid.py
  18. 1
      mmcv/cnn/bricks/hswish.py
  19. 1
      mmcv/cnn/bricks/non_local.py
  20. 1
      mmcv/cnn/bricks/norm.py
  21. 1
      mmcv/cnn/bricks/padding.py
  22. 1
      mmcv/cnn/bricks/registry.py
  23. 1
      mmcv/cnn/bricks/scale.py
  24. 1
      mmcv/cnn/bricks/swish.py
  25. 1
      mmcv/cnn/bricks/transformer.py
  26. 1
      mmcv/cnn/bricks/upsample.py
  27. 1
      mmcv/cnn/bricks/wrappers.py
  28. 1
      mmcv/cnn/builder.py
  29. 2
      mmcv/cnn/resnet.py
  30. 2
      mmcv/cnn/utils/__init__.py
  31. 1
      mmcv/cnn/utils/fuse_conv_bn.py
  32. 2
      mmcv/cnn/utils/weight_init.py
  33. 2
      mmcv/cnn/vgg.py
  34. 1
      mmcv/engine/__init__.py
  35. 1
      mmcv/engine/test.py
  36. 2
      mmcv/fileio/__init__.py
  37. 1
      mmcv/fileio/file_client.py
  38. 2
      mmcv/fileio/handlers/__init__.py
  39. 2
      mmcv/fileio/handlers/base.py
  40. 2
      mmcv/fileio/handlers/json_handler.py
  41. 2
      mmcv/fileio/handlers/pickle_handler.py
  42. 2
      mmcv/fileio/handlers/yaml_handler.py
  43. 2
      mmcv/fileio/io.py
  44. 2
      mmcv/fileio/parse.py
  45. 2
      mmcv/image/__init__.py
  46. 2
      mmcv/image/colorspace.py
  47. 2
      mmcv/image/geometric.py
  48. 2
      mmcv/image/io.py
  49. 1
      mmcv/image/misc.py
  50. 1
      mmcv/image/photometric.py
  51. 1
      mmcv/onnx/__init__.py
  52. 1
      mmcv/onnx/info.py
  53. 1
      mmcv/onnx/onnx_utils/__init__.py
  54. 1
      mmcv/onnx/onnx_utils/symbolic_helper.py
  55. 1
      mmcv/onnx/symbolic.py
  56. 1
      mmcv/ops/__init__.py
  57. 1
      mmcv/ops/bbox.py
  58. 1
      mmcv/ops/border_align.py
  59. 1
      mmcv/ops/box_iou_rotated.py
  60. 1
      mmcv/ops/carafe.py
  61. 1
      mmcv/ops/cc_attention.py
  62. 1
      mmcv/ops/contour_expand.py
  63. 1
      mmcv/ops/corner_pool.py
  64. 1
      mmcv/ops/csrc/bbox_overlaps_cuda_kernel.cuh
  65. 1
      mmcv/ops/csrc/border_align_cuda_kernel.cuh
  66. 1
      mmcv/ops/csrc/carafe_cuda_kernel.cuh
  67. 1
      mmcv/ops/csrc/carafe_naive_cuda_kernel.cuh
  68. 1
      mmcv/ops/csrc/cc_attention_cuda_kernel.cuh
  69. 1
      mmcv/ops/csrc/deform_roi_pool_cuda_kernel.cuh
  70. 1
      mmcv/ops/csrc/masked_conv2d_cuda_kernel.cuh
  71. 1
      mmcv/ops/csrc/nms_cuda_kernel.cuh
  72. 1
      mmcv/ops/csrc/onnxruntime/corner_pool.h
  73. 1
      mmcv/ops/csrc/onnxruntime/cpu/corner_pool.cpp
  74. 1
      mmcv/ops/csrc/onnxruntime/cpu/gridSample.cpp
  75. 1
      mmcv/ops/csrc/onnxruntime/cpu/nms.cpp
  76. 1
      mmcv/ops/csrc/onnxruntime/cpu/onnxruntime_register.cpp
  77. 1
      mmcv/ops/csrc/onnxruntime/cpu/reduce_ops.cpp
  78. 1
      mmcv/ops/csrc/onnxruntime/cpu/roi_align.cpp
  79. 1
      mmcv/ops/csrc/onnxruntime/cpu/soft_nms.cpp
  80. 1
      mmcv/ops/csrc/onnxruntime/grid_sample.h
  81. 1
      mmcv/ops/csrc/onnxruntime/nms.h
  82. 1
      mmcv/ops/csrc/onnxruntime/onnxruntime_register.h
  83. 1
      mmcv/ops/csrc/onnxruntime/ort_mmcv_utils.h
  84. 1
      mmcv/ops/csrc/onnxruntime/reduce_ops.h
  85. 1
      mmcv/ops/csrc/onnxruntime/roi_align.h
  86. 1
      mmcv/ops/csrc/onnxruntime/roi_align_rotated.h
  87. 1
      mmcv/ops/csrc/onnxruntime/soft_nms.h
  88. 1
      mmcv/ops/csrc/parrots/bbox_overlaps.cpp
  89. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_cuda.cu
  90. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_parrots.cpp
  91. 1
      mmcv/ops/csrc/parrots/bbox_overlaps_pytorch.h
  92. 1
      mmcv/ops/csrc/parrots/border_align.cpp
  93. 1
      mmcv/ops/csrc/parrots/border_align_cuda.cu
  94. 1
      mmcv/ops/csrc/parrots/border_align_parrots.cpp
  95. 1
      mmcv/ops/csrc/parrots/border_align_pytorch.h
  96. 1
      mmcv/ops/csrc/parrots/box_iou_rotated_parrots.cpp
  97. 1
      mmcv/ops/csrc/parrots/box_iou_rotated_pytorch.h
  98. 1
      mmcv/ops/csrc/parrots/carafe.cpp
  99. 1
      mmcv/ops/csrc/parrots/carafe_cuda.cu
  100. 1
      mmcv/ops/csrc/parrots/carafe_naive.cpp
  101. Some files were not shown because too many files have changed in this diff Show More

@ -1,4 +1,4 @@
Copyright (c) Open-MMLab. All rights reserved. Copyright (c) OpenMMLab. All rights reserved
Apache License Apache License
Version 2.0, January 2004 Version 2.0, January 2004

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
# flake8: noqa # flake8: noqa
from .arraymisc import * from .arraymisc import *
from .fileio import * from .fileio import *

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .quantization import dequantize, quantize from .quantization import dequantize, quantize
__all__ = ['quantize', 'dequantize'] __all__ = ['quantize', 'dequantize']

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import numpy as np import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .alexnet import AlexNet from .alexnet import AlexNet
# yapf: disable # yapf: disable
from .bricks import (ACTIVATION_LAYERS, CONV_LAYERS, NORM_LAYERS, from .bricks import (ACTIVATION_LAYERS, CONV_LAYERS, NORM_LAYERS,

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import logging import logging
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .activation import build_activation_layer from .activation import build_activation_layer
from .context_block import ContextBlock from .context_block import ContextBlock
from .conv import build_conv_layer from .conv import build_conv_layer

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
from torch import nn from torch import nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from torch import nn from torch import nn
from .registry import CONV_LAYERS from .registry import CONV_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math import math
from torch import nn from torch import nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import warnings import warnings
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
from .conv_module import ConvModule from .conv_module import ConvModule

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math import math
import numpy as np import numpy as np

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
from .registry import ACTIVATION_LAYERS from .registry import ACTIVATION_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
from .registry import ACTIVATION_LAYERS from .registry import ACTIVATION_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from abc import ABCMeta from abc import ABCMeta
import torch import torch

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import inspect import inspect
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
from .registry import PADDING_LAYERS from .registry import PADDING_LAYERS

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from mmcv.utils import Registry from mmcv.utils import Registry
CONV_LAYERS = Registry('conv layer') CONV_LAYERS = Registry('conv layer')

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import copy import copy
import warnings import warnings

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
r"""Modified from https://github.com/facebookresearch/detectron2/blob/master/detectron2/layers/wrappers.py # noqa: E501 r"""Modified from https://github.com/facebookresearch/detectron2/blob/master/detectron2/layers/wrappers.py # noqa: E501
Wrap some nn modules to support empty tensor input. Currently, these wrappers Wrap some nn modules to support empty tensor input. Currently, these wrappers

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..runner import Sequential from ..runner import Sequential
from ..utils import Registry, build_from_cfg from ..utils import Registry, build_from_cfg

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import logging import logging
import torch.nn as nn import torch.nn as nn

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .flops_counter import get_model_complexity_info from .flops_counter import get_model_complexity_info
from .fuse_conv_bn import fuse_conv_bn from .fuse_conv_bn import fuse_conv_bn
from .weight_init import (INITIALIZERS, Caffe2XavierInit, ConstantInit, from .weight_init import (INITIALIZERS, Caffe2XavierInit, ConstantInit,

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import copy import copy
import math import math
import warnings import warnings

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import logging import logging
import torch.nn as nn import torch.nn as nn

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .test import (collect_results_cpu, collect_results_gpu, multi_gpu_test, from .test import (collect_results_cpu, collect_results_gpu, multi_gpu_test,
single_gpu_test) single_gpu_test)

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp import os.path as osp
import pickle import pickle
import shutil import shutil

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .file_client import BaseStorageBackend, FileClient from .file_client import BaseStorageBackend, FileClient
from .handlers import BaseFileHandler, JsonHandler, PickleHandler, YamlHandler from .handlers import BaseFileHandler, JsonHandler, PickleHandler, YamlHandler
from .io import dump, load, register_handler from .io import dump, load, register_handler

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import inspect import inspect
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
from urllib.request import urlopen from urllib.request import urlopen

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .base import BaseFileHandler from .base import BaseFileHandler
from .json_handler import JsonHandler from .json_handler import JsonHandler
from .pickle_handler import PickleHandler from .pickle_handler import PickleHandler

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import json import json
import numpy as np import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import pickle import pickle
from .base import BaseFileHandler from .base import BaseFileHandler

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import yaml import yaml
try: try:

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from pathlib import Path from pathlib import Path
from ..utils import is_list_of, is_str from ..utils import is_list_of, is_str

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
def list_from_file(filename, prefix='', offset=0, max_num=0, encoding='utf-8'): def list_from_file(filename, prefix='', offset=0, max_num=0, encoding='utf-8'):
"""Load a text file and parse the content as a list of strings. """Load a text file and parse the content as a list of strings.

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .colorspace import (bgr2gray, bgr2hls, bgr2hsv, bgr2rgb, bgr2ycbcr, from .colorspace import (bgr2gray, bgr2hls, bgr2hsv, bgr2rgb, bgr2ycbcr,
gray2bgr, gray2rgb, hls2bgr, hsv2bgr, imconvert, gray2bgr, gray2rgb, hls2bgr, hsv2bgr, imconvert,
rgb2bgr, rgb2gray, rgb2ycbcr, ycbcr2bgr, ycbcr2rgb) rgb2bgr, rgb2gray, rgb2ycbcr, ycbcr2bgr, ycbcr2rgb)

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import cv2 import cv2
import numpy as np import numpy as np

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import numbers import numbers
import cv2 import cv2

@ -1,4 +1,4 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import io import io
import os.path as osp import os.path as osp
from pathlib import Path from pathlib import Path

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np import numpy as np
import mmcv import mmcv

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import cv2 import cv2
import numpy as np import numpy as np

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .info import is_custom_op_loaded from .info import is_custom_op_loaded
from .symbolic import register_extra_symbolics from .symbolic import register_extra_symbolics

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os import os

@ -0,0 +1 @@
# Copyright (c) OpenMMLab. All rights reserved.

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
"""Modified from https://github.com/pytorch/pytorch.""" """Modified from https://github.com/pytorch/pytorch."""
import warnings import warnings
from functools import wraps from functools import wraps

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
"""Modified from https://github.com/pytorch/pytorch.""" """Modified from https://github.com/pytorch/pytorch."""
import os import os

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from .bbox import bbox_overlaps from .bbox import bbox_overlaps
from .border_align import BorderAlign, border_align from .border_align import BorderAlign, border_align
from .box_iou_rotated import box_iou_rotated from .box_iou_rotated import box_iou_rotated

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..utils import ext_loader from ..utils import ext_loader
ext_module = ext_loader.load_ext('_ext', ['bbox_overlaps']) ext_module = ext_loader.load_ext('_ext', ['bbox_overlaps'])

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
# modified from # modified from
# https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/border_align.py # https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/border_align.py

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
from ..utils import ext_loader from ..utils import ext_loader
ext_module = ext_loader.load_ext('_ext', ['box_iou_rotated']) ext_module = ext_loader.load_ext('_ext', ['box_iou_rotated'])

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np import numpy as np
import torch import torch

@ -1,3 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
from torch import nn from torch import nn
from torch.autograd import Function from torch.autograd import Function

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BBOX_OVERLAPS_CUDA_KERNEL_CUH #ifndef BBOX_OVERLAPS_CUDA_KERNEL_CUH
#define BBOX_OVERLAPS_CUDA_KERNEL_CUH #define BBOX_OVERLAPS_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
// modified from // modified from
// https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/csrc/border_align/border_align_kernel.cu. // https://github.com/Megvii-BaseDetection/cvpods/blob/master/cvpods/layers/csrc/border_align/border_align_kernel.cu.
// the main difference: (1) use `argmax_idx` for fast computing of gradient // the main difference: (1) use `argmax_idx` for fast computing of gradient

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CARAFE_CUDA_KERNEL_CUH #ifndef CARAFE_CUDA_KERNEL_CUH
#define CARAFE_CUDA_KERNEL_CUH #define CARAFE_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CARAFE_NAIVE_CUDA_KERNEL_CUH #ifndef CARAFE_NAIVE_CUDA_KERNEL_CUH
#define CARAFE_NAIVE_CUDA_KERNEL_CUH #define CARAFE_NAIVE_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef CC_ATTENTION_CUDA_KERNEL_CUH #ifndef CC_ATTENTION_CUDA_KERNEL_CUH
#define CC_ATTENTION_CUDA_KERNEL_CUH #define CC_ATTENTION_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef DEFORM_ROI_POOL_CUDA_KERNEL_CUH #ifndef DEFORM_ROI_POOL_CUDA_KERNEL_CUH
#define DEFORM_ROI_POOL_CUDA_KERNEL_CUH #define DEFORM_ROI_POOL_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef MASKED_CONV2D_CUDA_KERNEL_CUH #ifndef MASKED_CONV2D_CUDA_KERNEL_CUH
#define MASKED_CONV2D_CUDA_KERNEL_CUH #define MASKED_CONV2D_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef NMS_CUDA_KERNEL_CUH #ifndef NMS_CUDA_KERNEL_CUH
#define NMS_CUDA_KERNEL_CUH #define NMS_CUDA_KERNEL_CUH

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_CORNER_POOL_H #ifndef ONNXRUNTIME_CORNER_POOL_H
#define ONNXRUNTIME_CORNER_POOL_H #define ONNXRUNTIME_CORNER_POOL_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "corner_pool.h" #include "corner_pool.h"
#include "../ort_mmcv_utils.h" #include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <cmath> #include <cmath>
#include "../ort_mmcv_utils.h" #include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "nms.h" #include "nms.h"
#include <assert.h> #include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "onnxruntime_register.h" #include "onnxruntime_register.h"
#include "corner_pool.h" #include "corner_pool.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "reduce_ops.h" #include "reduce_ops.h"
#include <assert.h> #include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "roi_align.h" #include "roi_align.h"
#include "../ort_mmcv_utils.h" #include "../ort_mmcv_utils.h"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "soft_nms.h" #include "soft_nms.h"
#include <assert.h> #include <assert.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_GRIDSAMPLE_H #ifndef ONNXRUNTIME_GRIDSAMPLE_H
#define ONNXRUNTIME_GRIDSAMPLE_H #define ONNXRUNTIME_GRIDSAMPLE_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_NMS_H #ifndef ONNXRUNTIME_NMS_H
#define ONNXRUNTIME_NMS_H #define ONNXRUNTIME_NMS_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_REGISTER_H #ifndef ONNXRUNTIME_REGISTER_H
#define ONNXRUNTIME_REGISTER_H #define ONNXRUNTIME_REGISTER_H
#include <onnxruntime_c_api.h> #include <onnxruntime_c_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ORT_MMCV_UTILS_H #ifndef ORT_MMCV_UTILS_H
#define ORT_MMCV_UTILS_H #define ORT_MMCV_UTILS_H
#include <onnxruntime_cxx_api.h> #include <onnxruntime_cxx_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_REDUCE_OPS_H #ifndef ONNXRUNTIME_REDUCE_OPS_H
#define ONNXRUNTIME_REDUCE_OPS_H #define ONNXRUNTIME_REDUCE_OPS_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_ROI_ALIGN_H #ifndef ONNXRUNTIME_ROI_ALIGN_H
#define ONNXRUNTIME_ROI_ALIGN_H #define ONNXRUNTIME_ROI_ALIGN_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_ROI_ALIGN_ROTATED_H #ifndef ONNXRUNTIME_ROI_ALIGN_ROTATED_H
#define ONNXRUNTIME_ROI_ALIGN_ROTATED_H #define ONNXRUNTIME_ROI_ALIGN_ROTATED_H

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef ONNXRUNTIME_SOFT_NMS_H #ifndef ONNXRUNTIME_SOFT_NMS_H
#define ONNXRUNTIME_SOFT_NMS_H #define ONNXRUNTIME_SOFT_NMS_H
#include <onnxruntime_cxx_api.h> #include <onnxruntime_cxx_api.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp" #include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "bbox_overlaps_cuda_kernel.cuh" #include "bbox_overlaps_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp" #include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp> #include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp> #include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp> #include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BBOX_OVERLAPS_PYTORCH_H #ifndef BBOX_OVERLAPS_PYTORCH_H
#define BBOX_OVERLAPS_PYTORCH_H #define BBOX_OVERLAPS_PYTORCH_H
#include <torch/extension.h> #include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp" #include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "border_align_cuda_kernel.cuh" #include "border_align_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp" #include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp> #include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp> #include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp> #include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BORDER_ALIGN_PYTORCH_H #ifndef BORDER_ALIGN_PYTORCH_H
#define BORDER_ALIGN_PYTORCH_H #define BORDER_ALIGN_PYTORCH_H
#include <torch/extension.h> #include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include <parrots/compute/aten.hpp> #include <parrots/compute/aten.hpp>
#include <parrots/extension.hpp> #include <parrots/extension.hpp>
#include <parrots/foundation/ssattrs.hpp> #include <parrots/foundation/ssattrs.hpp>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#ifndef BOX_IOU_ROTATED_PYTORCH_H #ifndef BOX_IOU_ROTATED_PYTORCH_H
#define BOX_IOU_ROTATED_PYTORCH_H #define BOX_IOU_ROTATED_PYTORCH_H
#include <torch/extension.h> #include <torch/extension.h>

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp" #include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "carafe_cuda_kernel.cuh" #include "carafe_cuda_kernel.cuh"
#include "pytorch_cuda_helper.hpp" #include "pytorch_cuda_helper.hpp"

@ -1,3 +1,4 @@
// Copyright (c) OpenMMLab. All rights reserved
#include "pytorch_cpp_helper.hpp" #include "pytorch_cpp_helper.hpp"
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA

Some files were not shown because too many files have changed in this diff Show More

Loading…
Cancel
Save