public class WekaScoreCAM extends AbstractCNNSaliencyMapWrapper
-bs <int> The mini batch size to use for map generation
-target-classes <int,int,...> Output class to generate saliency maps for; default is -1 (use the highest probability class). This only needs to be set if wanting to use a non-default class from the *command line*; if using the *GUI*, the 'View Saliency Map' window contains the interface for setting this.
-output <file location> File for the saliency map to be saved in
-normalize When generating the heatmap, should the values be normalized to be in [0, 1]
Constructor and Description |
---|
WekaScoreCAM() |
Modifier and Type | Method and Description |
---|---|
java.awt.image.BufferedImage |
generateHeatmapToImage()
Generates heatmaps, returning a human-viewable heatmap summary.
|
void |
processImage(java.io.File imageFile)
Main processing entrypoint.
|
getBatchSize, getClassMap, getCustomModelSetup, getDl4jMlpClassifier, getNormalizeHeatmap, getOptions, getOutputFile, getProgressManager, getTargetClassIDs, getTargetClassIDsAsInt, listOptions, saveResult, setBatchSize, setClassMap, setCustomModelSetup, setDl4jMlpClassifier, setNormalizeHeatmap, setOptions, setOutputFile, setTargetClassIDs, setTargetClassIDsAsInt
public void processImage(java.io.File imageFile)
AbstractCNNSaliencyMapWrapper
processImage
in class AbstractCNNSaliencyMapWrapper
imageFile
- Image to process heatmap for.public java.awt.image.BufferedImage generateHeatmapToImage()
AbstractCNNSaliencyMapWrapper
generateHeatmapToImage
in class AbstractCNNSaliencyMapWrapper