Skip to content
Snippets Groups Projects
Commit f401ef87 authored by André Anjos's avatar André Anjos :speech_balloon: Committed by Daniel CARRON
Browse files

[engine/saliencymap_generator] Improve comments on code

parent 4d6e9c13
No related branches found
No related tags found
1 merge request!12Adds grad-cam support on classifiers
......@@ -203,7 +203,7 @@ def run(
_save_saliency_map(use_folder, name, saliency_map) # type: ignore
else:
# pytorch-grad-cam will evaluate the output with the
# pytorch-grad-cam will figure out the output with the
# highest value and produce a saliency map for it - we
# will save it to disk.
use_folder = (
......@@ -211,6 +211,9 @@ def run(
)
saliency_map = saliency_map_callable(
input_tensor=image,
# setting `targets=None` will set target to the
# maximum output index using
# ClassifierOutputTarget(max_output_index)
targets=None, # type: ignore
)
_save_saliency_map(use_folder, name, saliency_map) # type: ignore
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment