pytorch-grad-cam
pytorch-grad-cam copied to clipboard
How should I assign a value to the "targets" when I draw the heatmap of yolov8-pose?
trafficstars
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"