32 #include "torch/script.h" 39 const std::string& str) {
42 CoordinateMapping::BALL_TO_CUBE_RADIAL;
43 if (str ==
"ball_to_cube_radial") {
44 coordinate_mapping = CoordinateMapping::BALL_TO_CUBE_RADIAL;
45 }
else if (str ==
"ball_to_cube_volume_preserving") {
46 coordinate_mapping = CoordinateMapping::BALL_TO_CUBE_VOLUME_PRESERVING;
47 }
else if (str ==
"identity") {
48 coordinate_mapping = CoordinateMapping::IDENTITY;
51 "coordinate_mapping must be one of ('ball_to_cube_radial', " 52 "'ball_to_cube_volume_preserving', 'identity') but got " +
55 return coordinate_mapping;
59 const std::string& str) {
62 if (str ==
"linear") {
63 interpolation = InterpolationMode::LINEAR;
64 }
else if (str ==
"linear_border") {
65 interpolation = InterpolationMode::LINEAR_BORDER;
66 }
else if (str ==
"nearest_neighbor") {
70 "interpolation must be one of ('linear', " 71 "'linear_border', 'nearest_neighbor') but got " +
Definition: VoxelPooling.h:40
InterpolationMode
Definition: ContinuousConvTypes.h:37
CoordinateMapping
Definition: ContinuousConvTypes.h:45
open3d::ml::impl::CoordinateMapping ParseCoordinateMappingStr(const std::string &str)
Definition: ContinuousConvHelper.h:38
open3d::ml::impl::InterpolationMode ParseInterpolationStr(const std::string &str)
Definition: ContinuousConvHelper.h:58