A proof-of-concept pipeline to detect empty seats in common work areas.
Clone or download this repo
git@github.com:MDass/open-space-detectives.git
Add a .env file in this directory the following line:
OPENAI_API_KEY = "[YOUR OPEN AI KEY]"
To train a custom YOLOv8 model, see:
finetune_yolo.py
To run inference on the custom models we generated, see:
run_inference.ipynb
To see our Person-Chair Collision Algorithm that detects whether a a person's bounding box collides with a chair's bounding box given out-of-the-box YOLOv8 'person' and 'chair' labels, see:
person_chair_collision.py
Open Space Detectives was created by Michelle Liu and Megan Dass, which was the result of a class project at Stanford University.
The software is available under the MIT License.
If you have any questions, feel free to open an issue or contact Megan Dass or Michelle Liu.