Here is a demo of the system. When the system detects that a drill is picked up, it will automatically issue a safety glasses warning. To represent the presence of the safety glasses warnings, the border of the RGB image is colored red in the demo video. When the system detects that no drill is picked up, it will not issue any safety glasses warnings. To represent the absence of the safety glasses warnings, the border of the RGB image is colored green in the demo video. As shown in the demo video, the computer vision system successfully detects whether the operator picks up a drill.
Step 1: Hardware
I use wood (from Home Depot) to form a support structure. I then mount a Microsoft XBOX 360 Kinect Sensor (from Amazon) on the support structure to monitor the activity on the ground.
Step 2: Segmentation
An example consisting of an RGB image, a depth image and an image of the extracted object is shown.
It is challenging for a computer vision algorithm to determine whether the hand of the operator is holding a drill from the RGB image alone. However, with the depth information, the problem is easier.
My segmentation algorithm sets the color of a pixel on the RGB image to black if its corresponding depth is outside a predefined range. This enables me to segment the object that is picked up.
Step 3: Classification
I collect data by videotaping myself holding a drill/waving hands separately. I then use the technique of transfer learning to tune a VGG neural network that is pre-trained using ImageNet. But the result is not good. Perhaps the extracted images are not similar to the natural images in ImageNet. Therefore, I train a convolutional neutral network using the extracted images from scratch. The result is pretty good. The accuracy of the classifier is ~95% on the validation set. A snippet of the model is given in the .py file.
Step 4: Have Fun and Be Safe !
Every day about 2,000 U.S. workers sustain job-related eye injuries that require medical treatment.
Nearly 60% injured workers were not wearing eye protection at the time of the accident or were wearing the wrong kind of eye protection for the job.
Have fun and be safe
Safety should always come first. My heart sinks whenever I hear of accidents involving power tools. I hope this article can raise the awareness that artificial intelligence can offer us an extra level of protection.
Have fun making things and be safe !