-
mask-detection-app.py
: main file which is executed withstreamlit run mask-detection-app.py
-
input.py
: algorithm for predicting image input files, similar to forward function in PyTorch -
data.py
: data structures that contain file paths to sample images -
model.py
: functions for building and predicting with the object detection model
- clone the repo
pip install streamlit
streamlit run mask-detection-app.py