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

A Multi-Task Framework Using Mamba for Identity, Age, and Gender Classification from Hand Images

Amirabbas Rezasoltani - Alireza Hosseini - Ramin Toosi - MohammadAli Akhaee

Integration ‎of ‎E‎lectric Vehicles in Smart ‎Grid using ‎Dee‎p ‎Reinforcement ‎Learning

Farkhondeh Kiaee

دسته‌بندی متون خبری فارسی با یادگیری فعال

مینا طباطبائی - دکتر سعیده ممتازی

بررسی روش m-ary در تولید زنجیره‌های افزونه‌ کوتاه

هادی صادقی کاجی - دکتر زهرا کریمی - دکتر محمد غلامی

A Comparison between Slimed Network and Pruned Network for Head Pose Estimation

Amir Salimiparsa - Hadi Veisi - Mohammad-shahram Moin

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

مهسا رضائی - مرجان کائدی

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

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

Improving Drug-Target Interaction Prediction Using Enhanced Feature Selection

Maryam Taheri - Mohammad Reza Keyvanpour - Mohadeseh Saadat Mousavi

A New Routing Protocol in Internet of Vehicles Inspired of Spread Model of the Covid-19 Virus

Taha Yasin Rezapour - Esmaeil Zeinali - Reza Ebrahimi Atani - Mohammad Mehdi Gilanian Sadeghi

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

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

more

Samin Hamayesh - Version 41.3.1