Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
common.py 312 Bytes
Newer Older
maming's avatar
Initial commit  
maming committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 
# Copyright (c) 2025, NVIDIA CORPORATION.
# SPDX-License-Identifier: BSD-3-Clause

from enum import Enum
from typing import TypeVar

import PIL.Image
import torch

TTransform = TypeVar("TTransform", bound=torch.nn.Module)


class Sampling(Enum):
    BILINEAR = PIL.Image.BILINEAR
    NEAREST = PIL.Image.NEAREST