add LICENSE.md

vt-vl-lab · Oct 27, 2019 · 8b46292 · 8b46292
1 parent e0f74f4
commit 8b46292
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pytorch_CAM_v2.py b/pytorch_CAM_v2.py
@@ -1,3 +1,6 @@
+# original code from: https://github.com/metalbubble/CAM/blob/master/pytorch_CAM.py
+# modified by Jinwoo Choi (jinchoi@vt.edu)
+
 # simple implementation of CAM in PyTorch for the networks such as ResNet, DenseNet, SqueezeNet, Inception
 
 import io
@@ -151,4 +154,4 @@ def returnCAM(feature_conv, weight_softmax, class_idx):
     # cv2.imwrite(os.path.join(output_path, 'CAM_frame_{:05d}_pred_cls_{}.jpg'.format(frame_indices[0].cpu().numpy()[0], pred_cls)), result)
     # cv2.imwrite(os.path.join(output_path, 'CAM_frame_{:05d}_pred_cls_{}.jpg'.format(frame_indices[-1].cpu().numpy()[0], pred_cls)), result)
 
-print('Finished CAM visualization!')
+print('Finished CAM visualization!')