AkashDataScience commited on
Commit
dced210
·
1 Parent(s): cf9410f
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -66,11 +66,7 @@ def display_false_detection_data(false_detection_data, number_of_samples):
66
 
67
  def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True, num_false_detection_images=10):
68
  im0 = input_img.copy()
69
- im_resized = cv2.resize(im0, (640, 640))
70
- rgb_img = im_resized.copy()
71
- transforms = albumentations.Compose([ToTensorV2()])
72
- im_resized = transforms(image = im_resized)['image']
73
- im_resized = im_resized.unsqueeze(0)
74
  stride, names, pt = model.stride, model.names, model.pt
75
  imgsz = check_img_size((640, 640), s=stride) # check image size
76
 
@@ -118,9 +114,9 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
118
  misclassified_images = None
119
 
120
  cam = EigenCAM(model, target_layers)
121
- grayscale_cam = cam(im_resized)[0, :]
122
  cam_image = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True)
123
-
124
  return im0, cam_image, misclassified_images
125
 
126
  title = "YOLOv9 model to detect shirt/tshirt"
 
66
 
67
  def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True, num_false_detection_images=10):
68
  im0 = input_img.copy()
69
+ rgb_img = cv2.resize(im0, (640, 640))
 
 
 
 
70
  stride, names, pt = model.stride, model.names, model.pt
71
  imgsz = check_img_size((640, 640), s=stride) # check image size
72
 
 
114
  misclassified_images = None
115
 
116
  cam = EigenCAM(model, target_layers)
117
+ grayscale_cam = cam(im)[0, :]
118
  cam_image = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True)
119
+
120
  return im0, cam_image, misclassified_images
121
 
122
  title = "YOLOv9 model to detect shirt/tshirt"