fernandoperlar/preprocessing_image
0
1U
2 F�a3 � @ s: d dl mZ d dlZd dlZd dlmZ G dd� d�Z dS )� N)�
preprocessingc @ s e Zd Zdd� Zdd� ZdS )�Joinc G s3 || _ d S )N)�models)�selfr � r �[c:\Users\ferna\OneDrive\Trabajo\CEIEC\MammoEchoThermo\mammo-echo-thermo\src\scripts\Join.py�__init__ s z
Join.__init__c s& t jt| j�ddd�\� }� �d|j� d|j� �� |D ] }|�� q8|d �� � � }� fdd�|D �}t4|�D �]�\}}t � t �
|j�t j��d �d5 d }tj|dd �}| j| �|�}| j| �|j|�\}} | j| j�|�d }6|�d
| j| j�� � �� |jddd�} | d �|d � | d �d� | d �d� | d �| j| �|d tj df �d � | d �d� | d �d� | d �t j� |�� | d �t� |7d�� | d �d� | d �| � | d jdt� |8�� �t� |9�t!|j�k�rdndd� | d �d� qpt �"� d S )N� T)�nrows�ncols�constrained_layoutzImage: z , Class: r c s g | ]}� � |��qS r )�
add_subfigure)�.0�gs��figr r �10<listcomp> s z*Join.visualize_heatmap.<locals>.<listcomp>�float32g �?�� )�axiszModel: � )r11 r ZOriginal�off.�Filter� � z Predicted: �g�r)�color)#�plt�subplots�lenr �suptitle�image�category�remove�get_subplotspec� get_gridspec� enumerate�cv�cvtColor�imread�
COLOR_BGR2RGB�astype�np� expand_dims�compute_heatmap� get_heatmap�model�predict�name�title�imshow� set_titler �filter�newaxisr � array_to_img�round�argmax�int�show) r r" Z12model_rowsZ model_row�gridspec�row�img�heatmap� jet_heatmap�superimposed_img� predicted�axr r r �visualize_heatmap13 s6 14$ ,8zJoin.visualize_heatmapN)�__name__�15__module__� __qualname__r rF r r r r r s r )16�matplotlib.pyplot�pyplotr �cv2r( �numpyr- �kerasr r r r r r �<module> s