File size: 530 Bytes
e05eed1 98a67a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
// SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
#pragma once
#include <torch/torch.h>
torch::Tensor rrect_to_quads(torch::Tensor rrects, float cellSize);
torch::Tensor rrect_to_quads_backward(torch::Tensor rrects, torch::Tensor gradOutput);
torch::Tensor calc_poly_min_rrect(torch::Tensor vertices);
float get_rel_continuation_cos(torch::Tensor rrectA, torch::Tensor rrectB);
torch::Tensor get_poly_bounds_quad(torch::Tensor poly);
|