Object Classification on Video with DataRobot Visual AI
This AI Accelerator demonstrates how deep learning model trained and deployed with DataRobot platform can be used for object detection on the video stream (detection if person in front of camera wears glasses).
Build with Free TrialDataRobot Visual AI allows you to train deep learning models intended for the Computer Vision projects that are demanded by the different industries. The object detection (binary and multiclass classification) applied to image and video processing is one of the tasks that can be easily and efficiently implemented with the DataRobot Visual AI. You can also bring your own Computer Vision model and deploy it in DataRobot via Custom Model Workshop.
This AI Accelerator demonstrates how deep learning model trained and deployed with DataRobot platform can be used for object detection on the video stream (detection if person in front of camera wears glasses). The Elastic-Net Classifier (L2 / Binomial Deviance) along with Pretrained MobileNetV3-Small-Pruned Multi-Level Global Average Pooling Image Featurizer with no image augmentation have been used for this Accelerator. The dataset used for the training can be found here, it contains images for 2 classes: persons with glasses and persons without glasses. The full size of the original dataset is 13.3 GB, of the cropped dataset is 335.8 MB. The small sample of the cropped dataset (100 images for each class) has been used for this AI Accelerator. The video stream is captured with OpenCV Computer Vision library. The frontend part has been implemented as a Streamlit application. The user of this AI Accelerator is expected to be familiar with the training and deployment process with DataRobot.
Accelerator overview
This accelerator requires:
- DataRobot deployment ID of the deep learning model.
- Installed OpenCV Computer Vision library.
- Installed Streamlit framework.
The following steps outline the accelerator workflow.
- Install OpenCV (
pip install opencv-python) - Install Streamlit (
pip install streamlit) - Train and deploy a deep learning model with DataRobot
- Implement Streamlit app (the code source can be found in this repository)
- Run application
streamlit run app.py -- --deployment_id DEPLOYMENT_ID
Explore more AI Accelerators
-
HorizontalPrediction Intervals via Conformal Inference
This AI Accelerator demonstrates various ways for generating prediction intervals for any DataRobot model. The methods presented here are rooted in the area of conformal inference (also known as conformal prediction).
Learn More -
HorizontalReinforcement Learning in DataRobot
In this notebook, we implement a very simple model based on the Q-learning algorithm. This notebook is intended to show a basic form of RL that doesn't require a deep understanding of neural networks or advanced mathematics and how one might deploy such a model in DataRobot.
Learn More -
HorizontalDimensionality Reduction in DataRobot Using t-SNE
t-SNE (t-Distributed Stochastic Neighbor Embedding) is a powerful technique for dimensionality reduction that can effectively visualize high-dimensional data in a lower-dimensional space.
Learn More -
HorizontalMLFlow + DataRobot API for Tracking Experimentation
As illustrated below, you will use the orchestration notebook to design and run the experiment notebook, with permutations of parameters handled automatically. At the end of the experiments, copies of the experiment notebook will be available, with the outputs for each permutation for collaboration and reference.
Learn More