BhumikaMak commited on
Commit
ad21b0c
·
1 Parent(s): ddc3ea5

Fix:update access lyrs for yolo8.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ def process_image(image, yolo_versions=["yolov5"]):
78
  if yolo_version == "yolov5":
79
  target_layers = [model.model.model.model[-2]] # Assumes last layer is used for Grad-CAM
80
  else:
81
- target_layers = [model.model.backbone[-1]]
82
  # Run YOLO detection
83
  results = model([rgb_img])
84
  boxes, colors, names = parse_detections(results)
 
78
  if yolo_version == "yolov5":
79
  target_layers = [model.model.model.model[-2]] # Assumes last layer is used for Grad-CAM
80
  else:
81
+ target_layers = [model.model.model[-1]]
82
  # Run YOLO detection
83
  results = model([rgb_img])
84
  boxes, colors, names = parse_detections(results)