Spaces:
Sleeping
Sleeping
Commit
·
e45fc58
1
Parent(s):
d9291e5
WIP
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def fig2img(fig):
|
|
32 |
return img
|
33 |
|
34 |
def composite_predictions(img, processed_predictions):
|
35 |
-
keep = processed_predictions["labels"]
|
36 |
boxes = processed_predictions["boxes"][keep].tolist()
|
37 |
scores = processed_predictions["scores"][keep].tolist()
|
38 |
labels = processed_predictions["labels"][keep].tolist()
|
@@ -49,7 +49,7 @@ def composite_predictions(img, processed_predictions):
|
|
49 |
plt.axis("off")
|
50 |
img = fig2img(plt.gcf())
|
51 |
matplotlib.pyplot.close()
|
52 |
-
return img
|
53 |
|
54 |
def process(img):
|
55 |
inputs = extractor(images=img, return_tensors="pt")
|
@@ -60,5 +60,7 @@ def process(img):
|
|
60 |
# Composite image and prediction bounding boxes + labels prediction
|
61 |
return composite_predictions(img, processed[0])
|
62 |
|
63 |
-
demo = gr.Interface(fn=process, inputs=[gr.Image(source="webcam", streaming=True, type='pil')], outputs=["image"], live=True)
|
64 |
-
|
|
|
|
|
|
32 |
return img
|
33 |
|
34 |
def composite_predictions(img, processed_predictions):
|
35 |
+
keep = processed_predictions["labels"] >= 1 # only interested in people
|
36 |
boxes = processed_predictions["boxes"][keep].tolist()
|
37 |
scores = processed_predictions["scores"][keep].tolist()
|
38 |
labels = processed_predictions["labels"][keep].tolist()
|
|
|
49 |
plt.axis("off")
|
50 |
img = fig2img(plt.gcf())
|
51 |
matplotlib.pyplot.close()
|
52 |
+
return img, len(boxes)
|
53 |
|
54 |
def process(img):
|
55 |
inputs = extractor(images=img, return_tensors="pt")
|
|
|
60 |
# Composite image and prediction bounding boxes + labels prediction
|
61 |
return composite_predictions(img, processed[0])
|
62 |
|
63 |
+
demo = gr.Interface(fn=process, inputs=[gr.Image(source="webcam", streaming=True, tool=None, type='pil')], outputs=["image", gr.Label(label="Current Attendance")], live=True)
|
64 |
+
|
65 |
+
if __name__ == "__main__":
|
66 |
+
demo.launch()
|