mirror of https://github.com/YifanXu74/MQ-Det.git
22 lines
897 B
C++
22 lines
897 B
C++
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
|
#pragma once
|
|
#include <torch/extension.h>
|
|
|
|
|
|
at::Tensor ROIAlign_forward_cpu(const at::Tensor& input,
|
|
const at::Tensor& rois,
|
|
const float spatial_scale,
|
|
const int pooled_height,
|
|
const int pooled_width,
|
|
const int sampling_ratio);
|
|
|
|
|
|
at::Tensor nms_cpu(const at::Tensor& dets,
|
|
const at::Tensor& scores,
|
|
const float threshold);
|
|
|
|
|
|
std::pair<at::Tensor, at::Tensor> soft_nms_cpu(const at::Tensor& dets,
|
|
const at::Tensor& scores,
|
|
const float threshold,
|
|
const float sigma); |