Skip to content

Equipment monitoring

Rust and Corrosion Detection

Ensure the safety of employees by inspecting machine equipment for the presence of rust/corrosion.

Rust and corrosion detection


Visual inspection of industrial environments is a common requirement across heavy industries, such as transportation, construction, and shipbuilding, and typically requires qualified experts to perform the inspection. Inspection locations can often be remote or in adverse environments that put humans at risk, such as bridges, skyscrapers, and offshore oil rigs.

Many of these industries deal with huge metal surfaces and harsh environments. A common problem across these industries is metal corrosion and rust. Although corrosion and rust are used interchangeably across different industries (we also use the terms interchangeably in this post), these two phenomena are different.

Visionify’s AI Vision Model for Rust/Corrosion Detection is designed to detect instances of rust/corrosion if any in machine parts, manufacturing equipments etc.


VisionAI model's generated events would be:

  • Rust or corrosion event detected

Model Details


The dataset for this scenario is based on rust/corrosion detection algorithms. The dataset is made up of images and videos gathered from various sources where instances of rust were found. The dataset has been catalogued to ensure real-world situations. It has an even distribution of:

  • Variations of pieces of equipment
  • Different(indoor/outdoor) environments
  • Different rust severity
  • Variations in camera orientations
  • Using security camera feeds

Total number of images used was 5572.


The model is based off of the YOLOv5 algorithm. The model is trained on a custom dataset of images and videos. The model is trained based on the above dataset compiled by our team.

Model card

Dataset size Version Camera support Precision Recall mAP
2326 v5 Straight 79% 49% 56%

The model is adaptable enough to run on any edge computing device.

Scenario details

The business logic for this scenario is as follows:

  • We use existing camera feeds from the premises to monitor the equipments in the workplace.
  • VisionAI system is able to run on edge devices. It uses camera feeds for processing.
  • We detect instances of rust/corrosion if any in machine parts, manufacturing equipments.

To test this model & scenario, you can use the following steps:

  • Install the visionai package from PyPI

    $ pip install visionai
  • Test the scenario from your local web-cam

    $ visionai scenario test rust-detection
    Downloading models for scenario: rust-detection
    Model: rust-detection:
    Starting scenario: rust-detection..
  • You should be able to see the events generated on your console window with the detections of smoking/vaping event within the camera field of view.


VisionAI app is available at a Azure Market place, one can download and use it by following steps mentioned here

Training with custom data

The scenario is provided as part of our GPL-v3 package for VisionAI. If you wish to train this with custom datasets, please contact us and we can provide you with the training code. You can do custom training with your own datasets for free, as long as it complies with GPLv3 license (you give back the code to the community). If you are interested in a custom license, please contact us.

Contact Us

  • For technical issues, you can open a Github issue here.
  • For business inquiries, you can contact us through our website.