کنفرانس بین المللی فناوری اطلاعات و دانش

English

صفحه اصلی / سیزدهمین کنفرانس بین المللی فناوری اطلاعات و دانش

Multi-label Classification of Steel Surface Defects Using Transfer Learning and Vision Transformer

نویسندگان :

Amirhossein Komijani¹ Farzaneh Vafaeinezhad² Javad Khoramdel³ Yasamin Borhani⁴ Esmaeil Najafi⁵

1- دانشگاه صنعتی خواجه نصیرالدین طوسی 2- دانشگاه صنعتی خواجه نصیرالدین طوسی 3- دانشگاه تربیت مدرس 4- دانشگاه صنعتی خواجه نصیرالدین طوسی 5- دانشگاه صنعتی خواجه نصیرالدین طوسی

کلمات کلیدی :

Computer vision،multi-label classification،steel defect detection،deep learning

چکیده :

Steel is the most widely used metal in various industries, for instance, automotive, buildings, packaging, etc. Defect detection on steel surfaces is crucial for steel production companies. This paper proposes a multi-label classification algorithm based on deep learning methods. Several models were trained based on the Severstal dataset, namely MobileNet- V2, Xception, DenseNet121, ViT, and ResNet-50 with transfer learning. In addition, a base model is trained from scratch based on convolutional layers such as ResNet. Moreover, a ViT model is constructed according to the attention and transformer layers concept. At first, 91% for the weighted F1 score is obtained. By exploiting the weighted loss method, the weighted F1 score increased to 92%.

لیست مقالات

لیست مقالات بایگانی شده

طراحی نرم افزاری مبتنی بر واقعیت افزوده با کاربرد فروش عینک

مینا علیانژاد - نسترن زنجانی - زهرا عسکری نژاد امیری

An Improved Drone Detection Method Using Deep Learning for Augmentation Detection Speed

Mohammad Bahrami - Seyyed Amir Asghari - Mohammadreza Binesh Marvasti - Sajjad Ansaria

Target-driven Navigation of a Mobile Robot using an End-to-end Deep Learning Approach

Mohammad Matin Hosni - Ali Kheiri - Esmaeil Najafi

Epileptic Seizure Detection based on Statistical and Wavelet Features and Siamese Network

Zahra Hossein-Nejad - Mehdi Nasri

شناسایی جایگاه مالونیلاسیون در پروتئین‌ها با بهره‌گیری از استخراج ویژگی و تکنیک‌های پردازش زبان طبیعی

حنانه رجبیون - محمد قاسم زاده - وحید رنجبر بافقی

Predicting Suicide Risk in Adolescents with Random Forest for Unbalanced Data Management

Fatemeh Rabbani - Dr Behrooz Masoumi - Dr Mohammad Reza Keyvanpour

Fast Online Character Recognition Using a Novel Local-Global Feature Extraction Method

Ayoub Parvizi - Dr Mohammad Kazemifard - Ziba Imani

بهبود هزینه‌های تراکنش در معماری مدیریت زنجیره‌ی تامین مبتنی بر زنجیره‌ی بلوکی

مژگان نوروزی نژاد - دکتر زهرا موحدی مژگان نوروزی نژاد - زهرا موحدی -

Comparative Study of Deep Reinforcement Learning and Genetic Algorithm Approaches for IoT Machine Learning Job Deployment in Fog Computing

Amir Moazeni - Omid Bushehrian

A Comparative Evaluation of Machine Learning Models for Anomaly-Based IDS in IoT Networks

Seyed Amir Mousavi - Mostafa Sadeghi - Mohammad Sadeq Sirjani

بیشتر

ثمین همایش، سامانه مدیریت کنفرانس ها و جشنواره ها - نگارش 42.5.2