-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move coarse rasterization to new file
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
- Loading branch information
1 parent
f2c44e3
commit 62dbf37
Showing
8 changed files
with
534 additions
and
480 deletions.
There are no files selected for viewing
File renamed without changes.
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
/* | ||
* 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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.