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

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

لیست مقالات

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

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

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

A Joint Trajectory and Energy Harvesting Method for an UAV Enabled Disaster Response Network

Hosein Mohammadi Firozjae - Javad Zeraatkar Moghaddam - Mehrdad Ardebilipour

Classical-Quantum Multiple Access Wiretap Channel with Common Message: One-shot Rate Region

Hadi Aghaee - Dr Bahareh Akhbari

تحلیل سازه‌های موثر بر پذیرش فناوری بلاکچین و استفاده از آن در صنعت بیمه ایران با استفاده از تکنیک معادلات ساختاری (مطالعه موردی: شرکت کارگزاری رسمی بیمه زندگی خوب)

احسان هنری - آفرین اخوان

OENMOP: Loss-Aware 4×4 and 5×5 and Scalable Non‑blocking Optical Switches Designed for Odd-Even Routing Algorithm for Chip-Scale Interconnection Networks

Negin Bagheri Renani - Elham Yaghoubi - Mina Mohammadirad

Architectural Insights: Comparing Weight Stationary and Output Stationary Systolic Arrays for Efficient Computation

Mahdi Kalbasi

A Mathematical Optimization Approach for Preference Learning in Movie Recommender Systems with Shared Accounts

Milad Khademali - Fazlollah Aghamohammadi - Marjan Kaedi - Alireza Nasiri

Web Service Ranking based on QoS and Use Prefer

Seyed Hossein Siadat - Danial Ramezani - Fatemeh Ahani

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

Fatemeh Rabbani - Dr Behrooz Masoumi - Dr Mohammad Reza Keyvanpour

PersianRAG A Retrieval Augmented Generation System for Persian Language

Hossein Hosseini - Mohammad Sobhan Zare - Amir Hossein Mohammadi - Arefeh Kazemi - Zahra Zojaji - Mohammad Ali Nematbakhsh

بیشتر

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