torchvision_nms_kernel.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Soumith Chintala 2016
//
// SPDX-License-Identifier: BSD-3-Clause

 
#pragma once
 
#include <ATen/ATen.h>
 
namespace reusex::vision::torchvision {

at::Tensor nms_kernel(const at::Tensor &dets, const at::Tensor &scores,
                      double iou_threshold);
 
} // namespace reusex::vision::torchvision