dirty-dish-detector

A collaborative computer vision project by Brian La, Justin Kong, and Richard Gu

Instructions for Viewing Demo/Results

To run the YOLO demo

• run src/yolo/video_feed.py

To generate the YOLO confusion matrix values

• run src/yolo/yolo.py

To generate the ResNet confusion matrix values

• run src/resnet/resnet_test.py

Edit model train configurations in dataset_generation.py and main.py

Models are saved to .pkl files