Spaces:
Running
Running
Commit
·
ad21b0c
1
Parent(s):
ddc3ea5
Fix:update access lyrs for yolo8.
Browse files
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.
|
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)
|