thienphuc12339/SignLanguage
0
1a
2 v��fo � @ s� d dl Z d dlZd dlmZ d dlmZmZmZ ej e3e4ej d�dd�Z de j e
e5e6d�d d7�Ze
ed �d d
�Zdej e
e8e9eej d�dd�ZdS )� N)�mimsave)�Image� ImageDraw� ImageFont)�image�std�mean�returnc C s& | | | } | d � d�} | �dd�S )N�� �uint8r )�astype�clip�r r r � r �iD:\OneDrive\Organizations\VuonAI\Projects\vietnamese-sign-language-translation\src\visualization\utils.py�unnormalize_img s r �10sample.gif)� video_tensor� save_pathr r c C sT g }| D ]*}t |�ddd��� ||d�}|�|� qddi}t||dfi |�� |S )N� � r r �durationg �?ZGIF)r �permute�numpy�appendr )r r r r �framesZ video_frameZframe_unnormalizedZkargsr r r � save_as_gif
s � r )�gif_pathr c C s11 t | d�S )N)�filename)r )r r r r � display_gif s r �� r! �r r r12 r! )r �text�position�color� font_sizer c C s> t jd|d�}t�| �}t�|�}|j||||d� t�|�S )Nzfonts/OpenSans-Regular.ttf)�font�size)�xyr# �fillr' ) r Ztruetyper � fromarrayr ZDrawr# �np�array)r r# r$ r% r& r' Z pil_image�drawr r r �draw_text_on_image$ s �1314�r/ )r NN)r r" r! )�torchr r, Zimageior �PILr r r �ndarray�tupler �Tensor�strr r �intr/ r r r r �<module> s2 �� ��