search

vmazashvili / Neural-Networks

Masktune Project reimplementation
0 stars 0 forks source link

call `plot_saliency_and_mask` after finetuning. #1

Closed vmazashvili closed 3 months ago

vmazashvili commented 3 months ago

call plot_saliency_and_mask after finetuning so that I can output the activation map for the masked image too

vmazashvili commented 3 months ago

let generate_saliency_map() return the classes too. this way, I can grab these classes for the class_idx in saliency_map = cam(class_idx=[sample_class], scores=output) for the new function generate_saliency_map_single()