Gradio

Step 1: Input Image

Step 2: BoundingBoxSelectAgent

BoundingBoxSelectAgent(classes=["person's head"], model=OwlV2)

Step 3: NMSAgent

NMSAgent(iou_threshold=0.5, score_threshold=0)

Step 4: SplitAgent

SplitAgent()

Step 5: ClassificationAgent

ClassificationAgent(classes=['hard hat', 'no hard hat'], model=CLIP)

Step 6: ClassMapAgent

ClassMapAgent(name_map={'hard hat': 'has ppe', 'no hard hat': 'no ppe'})

Step 7: JoinAgent

JoinAgent()