mirror of https://github.com/open-mmlab/mmcv.git
[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 OpenMMLabpull/1206/head^2
parent
04daea425b
commit
c47c9196d0
2
LICENSE
2
LICENSE
|
@ -1,4 +1,4 @@
|
|||
Copyright (c) Open-MMLab. All rights reserved.
|
||||
Copyright (c) OpenMMLab. All rights reserved
|
||||
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
# flake8: noqa
|
||||
from .arraymisc 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
|
||||
|
||||
__all__ = ['quantize', 'dequantize']
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
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
|
||||
# yapf: disable
|
||||
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 torch.nn as nn
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from .activation import build_activation_layer
|
||||
from .context_block import ContextBlock
|
||||
from .conv import build_conv_layer
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
from torch import nn
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from torch import nn
|
||||
|
||||
from .registry import CONV_LAYERS
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import math
|
||||
|
||||
from torch import nn
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import warnings
|
||||
|
||||
import torch.nn as nn
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch.nn as nn
|
||||
|
||||
from .conv_module import ConvModule
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import math
|
||||
|
||||
import numpy as np
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch.nn as nn
|
||||
|
||||
from .registry import ACTIVATION_LAYERS
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch.nn as nn
|
||||
|
||||
from .registry import ACTIVATION_LAYERS
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from abc import ABCMeta
|
||||
|
||||
import torch
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import inspect
|
||||
|
||||
import torch.nn as nn
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch.nn as nn
|
||||
|
||||
from .registry import PADDING_LAYERS
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from mmcv.utils import Registry
|
||||
|
||||
CONV_LAYERS = Registry('conv layer')
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import copy
|
||||
import warnings
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch.nn as nn
|
||||
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
|
||||
|
||||
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 ..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 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 .fuse_conv_bn import fuse_conv_bn
|
||||
from .weight_init import (INITIALIZERS, Caffe2XavierInit, ConstantInit,
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
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 math
|
||||
import warnings
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import logging
|
||||
|
||||
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,
|
||||
single_gpu_test)
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import os.path as osp
|
||||
import pickle
|
||||
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 .handlers import BaseFileHandler, JsonHandler, PickleHandler, YamlHandler
|
||||
from .io import dump, load, register_handler
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import inspect
|
||||
from abc import ABCMeta, abstractmethod
|
||||
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 .json_handler import JsonHandler
|
||||
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
|
||||
|
||||
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import json
|
||||
|
||||
import numpy as np
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import pickle
|
||||
|
||||
from .base import BaseFileHandler
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import yaml
|
||||
|
||||
try:
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from pathlib import Path
|
||||
|
||||
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'):
|
||||
"""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,
|
||||
gray2bgr, gray2rgb, hls2bgr, hsv2bgr, imconvert,
|
||||
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 numpy as np
|
||||
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import numbers
|
||||
|
||||
import cv2
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright (c) Open-MMLab. All rights reserved.
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import io
|
||||
import os.path as osp
|
||||
from pathlib import Path
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import numpy as np
|
||||
|
||||
import mmcv
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import cv2
|
||||
import numpy as np
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from .info import is_custom_op_loaded
|
||||
from .symbolic import register_extra_symbolics
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
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."""
|
||||
import warnings
|
||||
from functools import wraps
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
"""Modified from https://github.com/pytorch/pytorch."""
|
||||
import os
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from .bbox import bbox_overlaps
|
||||
from .border_align import BorderAlign, border_align
|
||||
from .box_iou_rotated import box_iou_rotated
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from ..utils import ext_loader
|
||||
|
||||
ext_module = ext_loader.load_ext('_ext', ['bbox_overlaps'])
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
# modified from
|
||||
# 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
|
||||
|
||||
ext_module = ext_loader.load_ext('_ext', ['box_iou_rotated'])
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import numpy as np
|
||||
import torch
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import torch
|
||||
from torch import nn
|
||||
from torch.autograd import Function
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef BBOX_OVERLAPS_CUDA_KERNEL_CUH
|
||||
#define BBOX_OVERLAPS_CUDA_KERNEL_CUH
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
// modified from
|
||||
// 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
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef 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
|
||||
#define CARAFE_NAIVE_CUDA_KERNEL_CUH
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef 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
|
||||
#define DEFORM_ROI_POOL_CUDA_KERNEL_CUH
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef 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
|
||||
#define NMS_CUDA_KERNEL_CUH
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_CORNER_POOL_H
|
||||
#define ONNXRUNTIME_CORNER_POOL_H
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "corner_pool.h"
|
||||
|
||||
#include "../ort_mmcv_utils.h"
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include <cmath>
|
||||
|
||||
#include "../ort_mmcv_utils.h"
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "nms.h"
|
||||
|
||||
#include <assert.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "onnxruntime_register.h"
|
||||
|
||||
#include "corner_pool.h"
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "reduce_ops.h"
|
||||
|
||||
#include <assert.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "roi_align.h"
|
||||
|
||||
#include "../ort_mmcv_utils.h"
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "soft_nms.h"
|
||||
|
||||
#include <assert.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_GRIDSAMPLE_H
|
||||
#define ONNXRUNTIME_GRIDSAMPLE_H
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_NMS_H
|
||||
#define ONNXRUNTIME_NMS_H
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_REGISTER_H
|
||||
#define ONNXRUNTIME_REGISTER_H
|
||||
#include <onnxruntime_c_api.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ORT_MMCV_UTILS_H
|
||||
#define ORT_MMCV_UTILS_H
|
||||
#include <onnxruntime_cxx_api.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_REDUCE_OPS_H
|
||||
#define ONNXRUNTIME_REDUCE_OPS_H
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef 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
|
||||
#define ONNXRUNTIME_ROI_ALIGN_ROTATED_H
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef ONNXRUNTIME_SOFT_NMS_H
|
||||
#define ONNXRUNTIME_SOFT_NMS_H
|
||||
#include <onnxruntime_cxx_api.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "pytorch_cpp_helper.hpp"
|
||||
|
||||
#ifdef MMCV_WITH_CUDA
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "bbox_overlaps_cuda_kernel.cuh"
|
||||
#include "pytorch_cuda_helper.hpp"
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include <parrots/compute/aten.hpp>
|
||||
#include <parrots/extension.hpp>
|
||||
#include <parrots/foundation/ssattrs.hpp>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef BBOX_OVERLAPS_PYTORCH_H
|
||||
#define BBOX_OVERLAPS_PYTORCH_H
|
||||
#include <torch/extension.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "pytorch_cpp_helper.hpp"
|
||||
|
||||
#ifdef MMCV_WITH_CUDA
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "border_align_cuda_kernel.cuh"
|
||||
#include "pytorch_cuda_helper.hpp"
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include <parrots/compute/aten.hpp>
|
||||
#include <parrots/extension.hpp>
|
||||
#include <parrots/foundation/ssattrs.hpp>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef BORDER_ALIGN_PYTORCH_H
|
||||
#define BORDER_ALIGN_PYTORCH_H
|
||||
#include <torch/extension.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include <parrots/compute/aten.hpp>
|
||||
#include <parrots/extension.hpp>
|
||||
#include <parrots/foundation/ssattrs.hpp>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#ifndef BOX_IOU_ROTATED_PYTORCH_H
|
||||
#define BOX_IOU_ROTATED_PYTORCH_H
|
||||
#include <torch/extension.h>
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "pytorch_cpp_helper.hpp"
|
||||
|
||||
#ifdef MMCV_WITH_CUDA
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "carafe_cuda_kernel.cuh"
|
||||
#include "pytorch_cuda_helper.hpp"
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
// Copyright (c) OpenMMLab. All rights reserved
|
||||
#include "pytorch_cpp_helper.hpp"
|
||||
|
||||
#ifdef MMCV_WITH_CUDA
|
||||
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue