Hugging Face's logo

Spaces:
HemaAM
/
YoloV3_PascalVOC_Dataset
Sleeping

App Files Community
HemaAM commited on
Commit
8905cfb
·
1 Parent(s): bc40ea8

Fixed bug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def detect_objects(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0
47
  if not enable_grad_cam:
48
  return [plot_img_with_bboxes]
49
 
50
- grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
51
  img = cv2.resize(image, (416, 416))
52
  img = np.float32(img) / 255
53
  grad_cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True, image_weight=transparency)
 
47
  if not enable_grad_cam:
48
  return [plot_img_with_bboxes]
49
 
50
+ grayscale_cam = yolo_grad_cam(transformed_image, scaled_anchors)[0, :, :]
51
  img = cv2.resize(image, (416, 416))
52
  img = np.float32(img) / 255
53
  grad_cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True, image_weight=transparency)