import torch
import numpy as np
import cv2
from typing import Union