stevfoy commited on
Commit
8e4c250
·
1 Parent(s): 9227d22
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,7 +68,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
68
  transformed_image = transforms(image=image)["image"].unsqueeze(0)
69
  print("transformed_image 2")
70
  output = model(transformed_image)
71
- print("transformed_image 3")
72
  bboxes = [[] for _ in range(1)]
73
 
74
  print("transformed_image 4")
@@ -88,7 +88,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
88
 
89
  print("transformed_image 6")
90
  grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
91
- img = cv2.resize(image, (416, 416))
92
  img = np.float32(img) / 255
93
  cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
94
 
 
68
  transformed_image = transforms(image=image)["image"].unsqueeze(0)
69
  print("transformed_image 2")
70
  output = model(transformed_image)
71
+ print("transformed_image 3")
72
  bboxes = [[] for _ in range(1)]
73
 
74
  print("transformed_image 4")
 
88
 
89
  print("transformed_image 6")
90
  grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
91
+ img = cv2.resize(image, (608, 608))
92
  img = np.float32(img) / 255
93
  cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
94