mirror of https://github.com/open-mmlab/mmcv.git
12 lines
356 B
C++
12 lines
356 B
C++
// Copyright (c) OpenMMLab. All rights reserved
|
|
#ifndef CONVEX_IOU_PYTORCH_H
|
|
#define CONVEX_IOU_PYTORCH_H
|
|
#include <torch/extension.h>
|
|
using namespace at;
|
|
|
|
void convex_iou(const Tensor pointsets, const Tensor polygons, Tensor ious);
|
|
|
|
void convex_giou(const Tensor pointsets, const Tensor polygons, Tensor output);
|
|
|
|
#endif // RIROI_ALIGN_ROTATED_PYTORCH_H
|