pytorch-grad-cam icon indicating copy to clipboard operation
pytorch-grad-cam copied to clipboard

How should I assign a value to the "targets" when I draw the heatmap of yolov8-pose?

Open Xavier-W opened this issue 1 year ago • 1 comments
trafficstars

image

I don't think "targets=None" is right. in the "base_cam.py",

if targets is None:
    target_categories = np.argmax(outputs.cpu().data.numpy(), axis=-2)
    targets = [ClassifierOutputTarget(
        category) for category in target_categories]

obviously, "pose" output should not use "ClassifierOutputTarget"

Xavier-W avatar Mar 15 '24 03:03 Xavier-W