BhumikaMak commited on
Commit
928f9f0
·
verified ·
1 Parent(s): 56801bb

cmap update

Browse files
Files changed (1) hide show
  1. yolov5.py +1 -1
yolov5.py CHANGED
@@ -2,7 +2,7 @@
2
  import torch
3
  import cv2
4
  import numpy as np
5
- from PIL import Image
6
  import torchvision.transforms as transforms
7
  from pytorch_grad_cam import EigenCAM
8
  from pytorch_grad_cam.utils.image import show_cam_on_image, scale_cam_image
 
2
  import torch
3
  import cv2
4
  import numpy as np
5
+ from PIL import Image
6
  import torchvision.transforms as transforms
7
  from pytorch_grad_cam import EigenCAM
8
  from pytorch_grad_cam.utils.image import show_cam_on_image, scale_cam_image