kopia lustrzana https://github.com/lzzcd001/MeshDiffusion
39 wiersze
896 B
C
39 wiersze
896 B
C
/*
|
|
* Copyright (c) 2020-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
*
|
|
* NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
|
|
* property and proprietary rights in and to this material, related
|
|
* documentation and any modifications thereto. Any use, reproduction,
|
|
* disclosure or distribution of this material and related documentation
|
|
* without an express license agreement from NVIDIA CORPORATION or
|
|
* its affiliates is strictly prohibited.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "common.h"
|
|
|
|
enum TonemapperType
|
|
{
|
|
TONEMAPPER_NONE = 0,
|
|
TONEMAPPER_LOG_SRGB = 1
|
|
};
|
|
|
|
enum LossType
|
|
{
|
|
LOSS_L1 = 0,
|
|
LOSS_MSE = 1,
|
|
LOSS_RELMSE = 2,
|
|
LOSS_SMAPE = 3
|
|
};
|
|
|
|
struct LossKernelParams
|
|
{
|
|
Tensor img;
|
|
Tensor target;
|
|
Tensor out;
|
|
dim3 gridSize;
|
|
TonemapperType tonemapper;
|
|
LossType loss;
|
|
};
|