34 lines
469 B
Python
34 lines
469 B
Python
from dataclasses import dataclass
|
|
from typing import Any, Dict
|
|
|
|
import torch
|
|
|
|
# Category
|
|
|
|
BASE_NODE_CATEGORY = "speech-dataset-toolkit"
|
|
|
|
# Value
|
|
|
|
MAX_SAFE_INT = 2**31 - 1 # 2,147,483,647
|
|
MAX_SAMPLE_RATE = 768000
|
|
|
|
# Type
|
|
|
|
AudioData = Dict[str, Any]
|
|
|
|
# Data
|
|
|
|
@dataclass(frozen=True)
|
|
class SpectrogramData:
|
|
"""
|
|
spectrogram data mainly handled for visualization
|
|
RETURN_TYPES = ("AUDIO", )
|
|
"""
|
|
|
|
waveform: torch.Tensor
|
|
|
|
sample_rate: int
|
|
|
|
|
|
# Node Pin
|