Apppublic

SV12/ERA_Session13

sourceHugging Facemitupdated 2y agoView on Hugging Face

0likes

rootutils__pycache__gradcam.cpython-310.pyc

gradcam.cpython-310.pyc22 linesDownload Raw Back to __pycache__

1o

26+fN3�@sVddlZddlmZddlmZddlmZddlm	Z4dd�Zdd�Zdd
d�Z
dS)�N)�GradCAM)�ClassifierOutputTarget)�show_cam_on_imagec
Cs�g}t||dd�}t|||�D].\}}}	t|���g}5||�d�|6dd�}|ddd�f}t|	|	��|dd�}|�|�q|S)NT)�model�
target_layers�use_cudar)�input_tensor�targetsZ7aug_smooth)�use_rgb)r�zipr�item�	unsqueezer�max�append)
rr�images�labelsZrgb_imgs�results�cam�image�labelZnp_imager	�
grayscale_cam�
visualization�r�>C:\Users\Shubham\Documents\ERA\ERA Session-13\utils\gradcam.py�generate_gradcam	s��rc8Cs�t�t|�dd�\}}|��t|��|||�D]"\}}}}	|�|�|�||�d||	���|�d�|�	�qt�9�dS)N�z | F)�plt�subplots�len�tight_layoutr�ravel�imshow�	set_title�grid�set_axis_off�show)10�misimgs�mistgts�mispreds�classes�fig�axes�ax�img�tgt�predrrr�visualize_gradcams111213r0�333333�?��14r3r�c	Cs�tj|d�}t||d�}t|�D]k}
t�|	|15|
d�||
d}|||d�}|ddd�f}|�d��d�}|dur?||�}t�|d�}|�	�}t16||d|d	�}t�|�t�d17|||
d�
��d|||
d�
����t�g�t�g�qdS)
N)�figsize)rr�r)rr	�cpu)r6rrT)r18Zimage_weightzLabel: z 19 Prediction: r)r�figurer�range�subplot�squeeze�to�np�	transpose�numpyrr!�titler�xticks�yticks)r�datar)r�number_of_samples�
inv_normalizer	�transparencyr5�rows�colsr*r�irrr-Zrgb_imgrrrr�plot_gradcam(s$20621�rJ)NNr1r2rr4)r?r=Zpytorch_grad_camrZ$pytorch_grad_cam.utils.model_targetsrZpytorch_grad_cam.utils.imager�matplotlib.pyplot�pyplotrrr0rJrrrr�<module>s22