mirror of
https://github.com/PyRetri/PyRetri.git
synced 2025-06-03 14:49:50 +08:00
31 lines
630 B
Python
31 lines
630 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from abc import abstractmethod
|
||
|
|
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
import torch
|
||
|
|
|
||
|
|
from ...utils import ModuleBase
|
||
|
|
from ...utils import Registry
|
||
|
|
|
||
|
|
from typing import Dict
|
||
|
|
|
||
|
|
|
||
|
|
class TransformerBase(ModuleBase):
|
||
|
|
"""
|
||
|
|
The base class of data augmentation operations.
|
||
|
|
"""
|
||
|
|
default_hyper_params = dict()
|
||
|
|
|
||
|
|
def __init__(self, hps: Dict or None = None):
|
||
|
|
"""
|
||
|
|
Args:
|
||
|
|
hps (dict): default hyper parameters in a dict (keys, values).
|
||
|
|
"""
|
||
|
|
super(TransformerBase, self).__init__(hps)
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def __call__(self, img: Image) -> Image or torch.tensor:
|
||
|
|
pass
|