vmazashvili / Neural-Networks

Masktune Project reimplementation
0 stars 0 forks source link

while running `generate_saliency_map`, sometimes RuntimeWarning: invalid value encountered in divide is encountered: #3

Closed vmazashvili closed 1 month ago

vmazashvili commented 1 month ago

<ipython-input-15-611cf68c253b>:3: RuntimeWarning: invalid value encountered in divide saliency_map_normalized = (saliency_map - np.min(saliency_map)) / (np.max(saliency_map) - np.min(saliency_map))

vmazashvili commented 1 month ago

Introduced Epsilon Clipping to the create_mask() function to avoid zero (unactive) regions in the maps:

epsilon = 1e-8 # A small value to avoid division by zero saliency_map_normalized = (saliency_map - np.min(saliency_map) + epsilon) / (np.max(saliency_map) - np.min(saliency_map) + epsilon)