RashiAgarwal
commited on
Commit
•
3b48481
1
Parent(s):
ba0213c
Update utils.py
Browse files
utils.py
CHANGED
@@ -661,7 +661,7 @@ class YoloCAM(BaseCAM):
|
|
661 |
if self.compute_input_gradient:
|
662 |
input_tensor = torch.autograd.Variable(input_tensor, requires_grad=True)
|
663 |
|
664 |
-
outputs = self.
|
665 |
if targets is None:
|
666 |
bboxes = [[] for _ in range(1)]
|
667 |
for i in range(3):
|
|
|
661 |
if self.compute_input_gradient:
|
662 |
input_tensor = torch.autograd.Variable(input_tensor, requires_grad=True)
|
663 |
|
664 |
+
outputs = self.activations_and_gradients(input_tensor)
|
665 |
if targets is None:
|
666 |
bboxes = [[] for _ in range(1)]
|
667 |
for i in range(3):
|