Spaces:
Running
on
Zero
Running
on
Zero
Update models/detection/detector.py
Browse files
models/detection/detector.py
CHANGED
@@ -61,7 +61,7 @@ class ObjectDetector:
|
|
61 |
})
|
62 |
return detections
|
63 |
|
64 |
-
|
65 |
overlay = image.copy()
|
66 |
draw = ImageDraw.Draw(overlay)
|
67 |
for det in detections:
|
|
|
61 |
})
|
62 |
return detections
|
63 |
|
64 |
+
def draw(self, image: Image.Image, detections, alpha=0.5):
|
65 |
overlay = image.copy()
|
66 |
draw = ImageDraw.Draw(overlay)
|
67 |
for det in detections:
|