Source code for draugr.torch_utilities.images.conversion

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 14-11-2020
           """
__all__ = ["quick_to_pil_image"]

import numpy
import torch
from PIL import Image

from draugr.opencv_utilities import to_rgb
from draugr.opencv_utilities.windows.image import show_image


[docs]def quick_to_pil_image(tensor: torch.Tensor, mode: str = "RGB") -> Image.Image: """ LOTS OF ASSUMPTIONS! :param tensor: :param mode: :return:""" return Image.fromarray(tensor.cpu().numpy().astype(numpy.uint8), mode)
if __name__ == "__main__": def asd2() -> None: """ :rtype: None """ import cv2 import torch from PIL import Image from tqdm import tqdm from draugr.opencv_utilities import frame_generator from draugr.torch_utilities import global_torch_device, to_tensor_generator with torch.no_grad(): for image in tqdm( to_tensor_generator( frame_generator(cv2.VideoCapture(0), coder=to_rgb), device=global_torch_device(), ) ): if show_image(numpy.array(quick_to_pil_image(image)), wait=1): break # esc to quit asd2()