mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-03 12:22:49 +08:00
62dbf371ae
Summary: In preparation for sharing coarse rasterization between point clouds and meshes, move the functions to a new file. No code changes. Reviewed By: bottler Differential Revision: D30367812 fbshipit-source-id: 9e73835a26c4ac91f5c9f61ff682bc8218e36c6a
39 lines
1.1 KiB
C++
39 lines
1.1 KiB
C++
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <torch/extension.h>
|
|
#include <tuple>
|
|
|
|
// Arguments are the same as RasterizeMeshesCoarse from
|
|
// rasterize_meshes/rasterize_meshes.h
|
|
#ifdef WITH_CUDA
|
|
torch::Tensor RasterizeMeshesCoarseCuda(
|
|
const torch::Tensor& face_verts,
|
|
const torch::Tensor& mesh_to_face_first_idx,
|
|
const torch::Tensor& num_faces_per_mesh,
|
|
const std::tuple<int, int> image_size,
|
|
const float blur_radius,
|
|
const int bin_size,
|
|
const int max_faces_per_bin);
|
|
#endif
|
|
|
|
// Arguments are the same as RasterizePointsCoarse from
|
|
// rasterize_points/rasterize_points.h
|
|
#ifdef WITH_CUDA
|
|
torch::Tensor RasterizePointsCoarseCuda(
|
|
const torch::Tensor& points,
|
|
const torch::Tensor& cloud_to_packed_first_idx,
|
|
const torch::Tensor& num_points_per_cloud,
|
|
const std::tuple<int, int> image_size,
|
|
const torch::Tensor& radius,
|
|
const int bin_size,
|
|
const int max_points_per_bin);
|
|
#endif
|