Spaces:
Sleeping
Sleeping
cmap update
Browse files
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
|