diff --git a/src/mednet/engine/saliency/viewer.py b/src/mednet/engine/saliency/viewer.py index 6e383801d10f0d0f4dc9875f85cc78cee0515084..7caae835739af74d8dac7baf3c38d987824b9350 100644 --- a/src/mednet/engine/saliency/viewer.py +++ b/src/mednet/engine/saliency/viewer.py @@ -63,7 +63,7 @@ def _overlay_saliency_map( Parameters ---------- image - The input image that will be overlayed with the saliency map. + The input image that will be overlaid with the saliency map. saliencies The saliency map that will be overlaid on the (raw) image. colormap @@ -118,7 +118,7 @@ def _overlay_bounding_box( Parameters ---------- image - The input image that will be overlayed with the saliency map. + The input image that will be overlaid with the saliency map. bbox The bounding box to draw on the input image. color @@ -149,13 +149,13 @@ def _process_sample( saliencies: numpy.typing.NDArray[numpy.double], ground_truth: BoundingBoxes, ) -> PIL.Image.Image: - """Generate an overlayed representation of the original sample and saliency + """Generate an overlaid representation of the original sample and saliency maps. Parameters ---------- raw_data - The raw data representing the input sample that will be overlayed with + The raw data representing the input sample that will be overlaid with saliency maps and annotations. saliencies The saliency map recovered from the model, that will be imprinted on @@ -166,7 +166,7 @@ def _process_sample( Returns ------- PIL.Image.Image - An image with the original raw data overlayed with the different + An image with the original raw data overlaid with the different elements as selected by the user. """