Professional Documents
Culture Documents
Data-Driven Pipe Object Detection and Classification For Enhanced Inventory Accuracy and Cost Reduction Using Artificial Intelligence Techniques
Data-Driven Pipe Object Detection and Classification For Enhanced Inventory Accuracy and Cost Reduction Using Artificial Intelligence Techniques
ISSN No:-2456-2165
Abstract:- Object detection has revolutionized industries Looking ahead, the paper investigates data-driven
like manufacturing, healthcare, and transportation by pipe detection and classification for enhanced inventory
enabling automated object identification and management. This promising approach utilizes
classification in images and videos. This paper explores computer vision algorithms to automate pipe
cutting-edge architectures like YOLOv3, YOLOv4 and identification and categorization, potentially
YOLOv8, highlighting their remarkable strides in revolutionizing inventory management practices and
accuracy, speed, and robustness. YOLOv8, with its streamlining operations.
balanced performance and ease of implementation, has
emerged as a leading architecture. However, practical By addressing YOLOv8 implementation challenges
challenges like overfitting, data collection difficulties, and exploring promising future directions, this paper
model complexity, and high hardware demands can contributes to the advancement of object detection
hinder its real-world adoption. technologies and their transformative impact across
various industries.
This paper investigates prominent toolboxes like
MMDetection and Detectron2 to address these Keywords:- Object Detection, YOLOv8, Pipe Inventory
challenges. Detectron2 provides optimization techniques Management, Image Processing, Artificial Intelligence,
and hardware acceleration strategies to tackle model Computer Vision, Deep Learning, Object Tracking, Real-
complexity and hardware demands. time Object Counting, Automated Inventory Management,
Pipe Classification.
The paper also explores deploying YOLOv8 on
Streamlit/Flask platforms, offering a user-friendly I. INTRODUCTION
interface for interacting with the model and visualizing
its detections, facilitating integration into web This article proposes a new automated pipe counting
applications. Strategies for overcoming deployment architecture based on the CRISP-ML(Q)
challenges and achieving optimal performance are also methodology[Fig.1]. The CNN architecture is designed to
discussed. Detect & classify the steel pipes in different shapes and
dimensions.
Fig 1 The CRISP-ML(Q) Architecture that we used for this Research can be Seen in the Figure Above
(Source: Mind Map - 360DigiTMG)
In the Business Understanding phase [Fig.1], we In the Modelling phase [Fig.3], we experimented with
identified the business problem of manually counting and different machine learning models [5]and model
tracking [10] the number of steel pipes of different architectures to identify the one that performs best on our
dimensions and shapes, which is a time-consuming and task of automated steel pipe detection and classification. We
error-prone process that can lead to substantial waste of time also augmented the training set with additional data to
and cost. Businesses can identify steel pipes of various sizes improve the model's generalization performance.
and shapes with high accuracy and efficiency by utilising
machine learning for automated steel pipe detection and Once the model is trained [6], we will evaluate its
classification. This reduces the likelihood of human mistake performance on a held-out test set to ensure that it meets our
and the need for manual intervention. requirements. If the model does not meet our requirements,
we will return to the Modelling phase and make further
In the Data Understanding phase, we explored and adjustments.
understood the dataset of steel pipe images with different
shapes and dimensions [11], and identified the need for data Once we are satisfied with the model's performance,
cleaning and preprocessing such as noise removal, resizing, we will deploy it to production [13] so that it can be used to
and image augmentation via techniques such as random detect and classify steel pipes in real-time. The model can be
cropping, flipping, and rotating images. integrated into a software application or with cameras.
In the data preparation phase [Fig.2], we pre-processed The proposed automated pipe detection system, which
and transformed the data by converting it into images in a follows the CRISP-ML(Q) [Fig.1] methodology and focuses
format of JPG that was suitable for model training. on modifying and using ML Workflow Architecture for
Additionally, we divided the data into test and training sets. machine learning models [Fig.2]., has the potential to
improve the speed, accuracy, and efficiency of object
detection, reduce the need for manual labour, and prevent
the shipment of incorrectly counted materials. This could
lead to significant cost savings and product loss reductions.
Fig 2 ML Workflow Architecture Used for the Research - A Detailed Overview of the Deep Learning Pipeline for Object
Detection and Classification
(Source : ML Workflow - 360DigiTMG)
Fig 3 Architecture Diagram: Showcasing the Components and Flow of Data for Automated Pipe Detection and Classification
using the CNN Machine Learning Technique
(Source: ML Workflow - 360DigiTMG)
As depicted in [Fig.3], Building and deploying a steps involve acquiring and preparing the data that the
YOLOv8-based object detection model involves a well- models will be trained on, ensuring their ability to
defined workflow that encompasses data preparation, model accurately identify and categorize pipes encountered in real-
training, evaluation, and deployment. Let's explore each world settings.
phase of this procedure:
The data [Table.1] employed in this study was sourced
Data Collection: directly from a warehouse specializing in steel pipes. This
Securing and annotating data form the critical comprehensive collection encompasses twenty-one distinct
foundation for constructing effective machine-learning categories of pipes, featuring diverse shapes such as square,
models for pipe detection and classification. These crucial rectangular, and circular configurations.
CNNs can learn to extract features from images that are Detectron2:
relevant for steel pipe detection and classification, such Detectron2[3] is a popular open-source object
as the presence of different types of steel pipes, defects, detection framework. Detectron2 provides a variety of pre-
and corrosion. trained models for object detection tasks, including steel
CNNs are robust to noise and other variations in the pipe detection and classification.
data.
CNNs can be trained to achieve very high accuracy on MM Detection:
steel pipe detection and classification tasks. MM Detection[7] is another popular open-source
object detection framework. MM Detection provides a
variety of pre-trained models for object detection tasks,
including steel pipe detection and classification.
The YOLOv8 [5] model's performance [Fig.5] was among all actual positives. The F1 score, a harmonic mean
assessed using a rigorous set of evaluation metrics, of precision and recall, provided a combined measure of
including precision, recall, F1 score, Intersection over Union model accuracy measured the overlap between predicted
(IoU) [Fig.6], and mean Average Precision (mAP)[Table.2]. bounding boxes and ground truth boxes, while mAP
Precision measured the proportion of correct positive provided a comprehensive summary of the model's overall
predictions among all positive predictions, while recall performance
measured the proportion of correct positive predictions
Fig 7 The Above Figure Depicts the Predicted Pipe Class and Counts using Streamlit Application