Spaces:
Sleeping
Sleeping
efraim1011
commited on
Commit
•
6e99ad0
1
Parent(s):
9ff5c58
Some categories removed
Browse files
app.py
CHANGED
@@ -212,7 +212,7 @@ def process_video(
|
|
212 |
|
213 |
def process_image_real_time(
|
214 |
input_image: np.ndarray,
|
215 |
-
categories: str = "safety helmet, safety glasses, hearing protectors, protective masks, safety shoes, protective gloves, seat belt, person, bicycle, car, motorcycle, airplane, dining table, bus, train, truck, boat,
|
216 |
confidence_threshold: float = 0.3,
|
217 |
iou_threshold: float = 0.5,
|
218 |
# with_segmentation: bool = True,
|
|
|
212 |
|
213 |
def process_image_real_time(
|
214 |
input_image: np.ndarray,
|
215 |
+
categories: str = "safety helmet, safety glasses, hearing protectors, protective masks, safety shoes, protective gloves, seat belt, person, bicycle, car, motorcycle, airplane, dining table, bus, train, truck, boat, parking meter, bench, bird, cat, dog, horse, elephant, bear, zebra, giraffe, backpack, umbrella, handbag, tie, suitcase, frisbee, skis, snowboard, sports ball, kite, baseball bat, baseball glove, skateboard, surfboard, tennis racket, bottle, wine glass, cup, fork, knife, spoon, bowl, banana, apple, sandwich, orange, donut, cake, chair, couch, potted plant, bed, toilet, tv, laptop, mouse, remote, keyboard, cell phone, microwave, oven, toaster, sink, refrigerator, book, clock, vase, teddy bear, hair drier, toothbrush",
|
216 |
confidence_threshold: float = 0.3,
|
217 |
iou_threshold: float = 0.5,
|
218 |
# with_segmentation: bool = True,
|