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

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%.

لیست مقالات

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

IT-based and Non-IT-based methods to separate and collect waste

Hoda Harati - Farzad Haghighi-Rad - Reza Yousefi Zenouz

تشخیص بیماری مزمن کلیوی با استفاده از یادگیرنده‌های گروهی و انتخاب ویژگی‌های مؤثر مبتنی‌ بر الگوریتم بهینه‌سازی تبادل حرارتی

صبا عارف‌نیا - مهدی هاشم‌زاده - امین گلزاری اسکوئی

Detection and Identification of Cyber-Attacks in Cyber-Physical Systems Based on Machine Learning Methods

Zohre Nasiri Zarandi

AI-based Message Spam Classification Framework for Secure Autonomous Vehicles Communication

Riya Upadhyay - Mili Virani - Lakshit Pathak - Rajesh Gupta - Sudeep Tanwar - Hossein Shahinzadeh

بکارگیری الگوریتم بهینه سازی فاخته و منطق فازی به منظور بهبود زمانبندی وظایف در محیط محاسبات مه

فاطمه دوامی - حمید جلیلوند - فاطمه نجفی

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

سحر فقیهی راد - دکتر سیده نفیسه آل محمد سحر فقیهی راد - سیده نفیسه آل محمد -

Mode Selection and Resource Allocation in D2D-Enabled MC-NOMA using Matching Theory

Alireza Gholamrezaee - Hamid Farrokhi - Javad Zeraatkar Moghaddam

روشی برای تشخیص مرحله پیشرفت آلزایمر در تصاویرFMRI مبتنی بر شبکه های عصبی چگال

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

تحلیل و بررسی تکنیک‌های محاسبات تقریبی

محمد میلاد صیاد - محمد رضا بینش مروستی - سید امیر اصغری

Open-domain question classification and completion in conversational information search

Omid Mohammadi Kia - Mahmood Neshati - Mahsa Soudi Alamdari

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