Skip to content

README.md

Latest commit

 

History

History
32 lines (26 loc) · 1.28 KB

README.md

File metadata and controls

32 lines (26 loc) · 1.28 KB

CV Final Project: Cat Eating Habits Tracker

Introduction

This project is a cat eating habits tracker. It is designed to help keep track of my cat's eating habits. My model is only trained to detect my cat, so it will not work for other cats, which is why I have provided sample data of my cats eating.

Preface

I could not gather organic sample data of my cats eating together in a short duration of time (would increase the sample video length by a lot) , so I had to manually stitch together the videos of my cats eating in all_cats.mp4.

Installation

I recommend using a virtual environment (venv) to install the necessary packages.

To install the necessary packages, run the following command:

pip install -r requirements.txt

Running Sample Data

To run the sample data, run the following command:

python detect_sample.py --input sample_data/grumpy.mp4
# or
python detect_sample.py --input sample_data/molly.mp4
# or
python detect_sample.py --input sample_data/tanner.mp4
# or
python detect_sample.py --input sample_data/all_cats.mp4

Output

The output video and csv file will be saved in the output directory. The output video will have bounding boxes around the detected cat(s) and the csv file will contain the timestamps of when the cat(s) were detected.