International Conference on Information and Knowledge Technology

فارسی

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

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

Authors :

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

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

Keywords :

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

Abstract :

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

List of archived papers

کشف برخط تقلب پیشنهاد ساختگی (Bid-Shielding) در مناقصه و مزایده‌های الکترونیکی هلندی با رویکرد تحلیل شبکه اجتماعی

فاطمه الثلایا - دکتر سید علیرضا هاشمی گلپایگانی فاطمه الثلایا - سید علیرضا هاشمی گلپایگانی -

Recommendation Systems in Smart Agriculture: Pathway to a well-designed system

Ahmad Nameni - Amir Ghafarian Daneshmand - Omid Mahdi Ebadati E

A novel approach audio watermarking based on (GBT,DCT,SVD)

Mahdi Mosleh

نقش داده‌های آنلاین یونیفرمیتی و تحلیل آماری پیشرفته با ترکیب پایتون و پاوربی‌آی در بهبود کیفیت و فرآیند تولید تایر

دانیال قادری

سیستم توصیه گر برای خرید لوازم آرایشی و بهداشتی مبتنی بر الگوریتم جنگل تصادفی

فاطمه رمضانی خوزستانی - مجید رفیعی

چارچوب بومی پیاده‌سازی حکمرانی داده در رسانه‌های عمومی بر پایه مدل EDM

مریم فتحی - عبدالله امیرخانی - فرشید بهجت محمدی - ملیحه حاجی حسینی

Particle Swarm Optimization-Based Framework for 3D Swarm Robotic Navigation Using Artificial Potential Field Dynamics

Samim Kamyab - Masoud Shirzadeh - Ghoncheh Zand

حفظ حریم خصوصی در انتشار نسخه های متوالی داده‌های شبکه اجتماعی با امکان افزایش یال

طاهره سرزهی - دکتر مهری رجایی طاهره سرزهی - مهری رجایی -

Vi-Net: A Deep Violent Flow Network for Violence Detection in Video Sequences

Tahereh Zarrat Ehsan - Seyed Mehdi Mohtavipour

Two Novel Designs of Efficient Single-Bit Comparators in QCA Technology with Ultra-Low Energy Dissipation

Shobeir Fayazi - Hatam Abdoli

more

Samin Hamayesh - Version 44.5.0