Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Grad-CAM numpy() on tensor with grad (add detach)

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-22 07:02:12 +03:00
parent b0e87c07ef
commit a32cb8ac6c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/training/gradcam.py
View File

@@ -146,7 +146,7 @@ class DINOv3GradCAM:
# Denormalize from ImageNet stats. # Denormalize from ImageNet stats.
mean = torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1).to(image.device) mean = torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1).to(image.device)
std = torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1).to(image.device) std = torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1).to(image.device)
img = (image * std + mean).clamp(0, 1).cpu().numpy().transpose(1, 2, 0) # [H, W, 3] img = (image * std + mean).clamp(0, 1).detach().cpu().numpy().transpose(1, 2, 0) # [H, W, 3]
# Resize heatmap to image size. # Resize heatmap to image size.
heatmap_resized = torch.tensor(heatmap).unsqueeze(0).unsqueeze(0).float() heatmap_resized = torch.tensor(heatmap).unsqueeze(0).unsqueeze(0).float()