[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 2021-08-08 07:10:55 -07:00 committed by GitHub
parent 04daea425b
commit c47c9196d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
339 changed files with 339 additions and 95 deletions

View File

@ -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

View File

@ -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 *

View File

@ -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']

View File

@ -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

View File

@ -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,

View File

@ -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

View File

@ -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

View File

@ -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

View File

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

View File

@ -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

View File

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

View File

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

View File

@ -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

View File

@ -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

View File

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

View File

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

View File

@ -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

View File

@ -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

View File

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

View File

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

View File

@ -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

View File

@ -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')

View File

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

View File

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

View File

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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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,

View File

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

View File

@ -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

View File

@ -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

View File

@ -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)

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

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

View File

@ -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

View File

@ -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.

View File

@ -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)

View File

@ -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

View File

@ -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

View File

@ -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

View File

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

View File

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

View File

@ -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

View File

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

View File

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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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'])

View File

@ -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

View File

@ -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'])

View File

@ -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

View File

@ -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

View File

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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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"

View File

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

View File

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

View File

@ -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"

View File

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

View File

@ -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"

View File

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

View File

@ -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

View File

@ -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

View File

@ -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>

View File

@ -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>

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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>

View File

@ -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

View File

@ -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"

View File

@ -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>

View File

@ -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>

View File

@ -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

View File

@ -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"

View File

@ -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>

View File

@ -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>

View File

@ -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>

View File

@ -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>

View File

@ -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

View File

@ -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"

View File

@ -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