0% Complete
English
صفحه اصلی
/
سیزدهمین کنفرانس بین المللی فناوری اطلاعات و دانش
Multi-label Classification of Steel Surface Defects Using Transfer Learning and Vision Transformer
نویسندگان :
Amirhossein Komijani
1
Farzaneh Vafaeinezhad
2
Javad Khoramdel
3
Yasamin Borhani
4
Esmaeil Najafi
5
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%.
لیست مقالات
لیست مقالات بایگانی شده
Analysing effect of news polarity on stock market prediction: a machine learning approach
Golshid Ranjbaran - Dr Mohammad-Shahram Moin - Dr Sasan H Alizadeh - Dr Abbas Koochari
An ESB-based Architecture for Authentication as a Service Through Enterprise Application Integration
Masoumeh Hashemi - Mehdi Sakhaei-nia - Morteza Yousef Sanati
AN EFFICIENT TASK SCHEDULING IN CLOUD COMPUTING BASED ON ACO ALGORITHM
Zahra Shafahi - Dr Alireza Yari
A Graph Attention-Based Autoencoder for Critical Path Anomaly Detection in Microservices
Mahdi Naderi - Hossein Momeni - Shayan Shahini
The risk prediction of heart disease by using neuro-fuzzy and improved GOA
Vahid Safari Dehnavi - Masoud Shafiee
Designing an AI-assisted toolbox for fitness activity recognition based on deep CNN
Ali Bidaran - Dr Saeed Sharifian
An approach to model the optimal service provisioning in vehicular cloud networks
Farhoud Jafari Kaleibar - Maghsoud Abbaspour
روشی برای بهبود آزمون جهش پیشگویانه با در نظر گرفتن اثر داده های از دست رفته
طه رستمی - دکتر سعید جلیلی طه رستمی - سعید جلیلی -
DynamicEvoStream : خوشه بندی پویای جریان داده تکاملی در زمانهای بیکاری
زهرا عمیقی - مرتضی یوسف صنعتی - میرحسین دزفولیان
طرحی برای تبدیل نمودارهای رفتاری BPMN به نمودار UML و تولید کد از آن
مهدیس صفری - احمد عبدالله زاده بارفروش
بیشتر
ثمین همایش، سامانه مدیریت کنفرانس ها و جشنواره ها - نگارش 41.3.1