Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -78,6 +78,9 @@ detector.prepare(-1, nms_thresh=0.5, input_size=(640, 640))
|
|
78 |
|
79 |
|
80 |
def detect(image: np.ndarray) -> tuple[Image.Image, list[Image.Image]]:
|
|
|
|
|
|
|
81 |
image = image[:, :, ::-1] # RGB -> BGR
|
82 |
bboxes, vbboxes = detect_person(image, detector)
|
83 |
res = visualize(image, bboxes, vbboxes)
|
|
|
78 |
|
79 |
|
80 |
def detect(image: np.ndarray) -> tuple[Image.Image, list[Image.Image]]:
|
81 |
+
if image is None:
|
82 |
+
return None, []
|
83 |
+
|
84 |
image = image[:, :, ::-1] # RGB -> BGR
|
85 |
bboxes, vbboxes = detect_person(image, detector)
|
86 |
res = visualize(image, bboxes, vbboxes)
|