Spaces:
Runtime error
Runtime error
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:
|