blanchon/Metric3D
0
1a
2 �6�d�3 � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dl Z d dl4Z5d dl m Z d dl
Z
d dlmZmZ de
je
jeeee
jd�dd�Zdee
je
je
jeed �d6d �Ze
je
je
jd �d
d�Zddd�ZdS )� N)�gray_to_colormap)� main_process)�Col�7imagetable� i@��pred�rgb�filename�save_dir�scale�targetc C s� t �tj�||dd� d �|� t �tj�||dd� d �| | �tj�� |dur�t �tj�||dd� d �|| �tj�� dS )z98 Save raw GT, predictions, RGB in the same file.9 N�����z_rgb.jpgz_d.pngz_gt.png)�cv2�imwrite�os�path�join�astype�np�uint16r � r �JC:\Users\JUGG\Documents\GitHub\Metric3DPrivate\mono\utils\visualization.py�
save_raw_imgs s ".r )�iterr r
r r10 r c C s� t |||�\}}}} }11|�d�}tj|| |12gdd�} t�tj�||dd� d �| � |dur�|� |dd� � d�| �d�| � dS )z513 Save GT, predictions, RGB in the same file.14 )� � r r )�axisNr z15_merge.jpg)r r r )16�get_data_for_log� transposer � concatenate�plt�imsaver r r � add_image) r r r
r r17 r Z tb_logger�18pred_scale� target_scale�19pred_color� target_colorZcat_imgr r r �
save_val_imgs s 20"r( )r r
r c 21 C s, t �g d��d d �t jt jf }t �g d��d d �t jt jf }| �� �� �� } |�� �� �� }|�� �� �� }d| | dk < d||dk < t| �� |�� �}| | d �t j�}|| d �t j�}t | �}t |�} t22� ||j d |j d f�}t23� | |j d |j d f�} || | �t j
�}||||| fS )N)g33333�^@gR� ��]@gR� ���Y@)g��(\�2M@g���(\�L@g �L@r i' r r )r �array�newaxis�squeeze�cpu�numpy�maxr r r r �resize�shape�uint8)24r r
r �mean�stdZ max_scaler$ r% r& r' r r r r 1 s r �25index.html�� i� c C s> g }| � � D ] \}}td||�}|�|� q t|||d� d S )N�img)�out_fileZimsize)�itemsr �appendr )Z name2path� save_path�size�cols�k�vZcol_ir r r � create_htmlG s26 r@ )r N)N)r4 r5 )�matplotlib.pyplot�pyplotr! r r r- r Zmono.utils.transformr �shutil�glob�mono.utils.runningr �torchZ html4visionr r �tensor�str�floatr �intr( r r@ r r r r �<module> s: �� ��