torch_inference.py How to output the desired image

[monkeycc]

https://github.com/openvinotoolkit/anomalib/blob/main/tools/inference/torch_inference.py

How to output the desired image

No text required
No blank space required

I want to directly obtain this type of image

[monkeycc]

\anomalib\src\anomalib\post_processing\visualizer.py

 elif self.task == TaskType.CLASSIFICATION:
 return image_result.heat_map

Problem solving
I have modified this