FINE TUNING DEEP LEARNING MODELS FOR PEDESTRIAN DETECTION

被引:8
|
作者
Amisse, Caisse [1 ,2 ]
Jijon-Palma, Mario Ernesto [1 ]
Silva Centeno, Jorge Antonio [1 ]
机构
[1] Univ Fed Parana, Programa Posgrad Ciencias Geodes, Curitiba, Parana, Brazil
[2] Univ Rovuma, Dept Ciencias Nat, Nampula, Mozambique
来源
BOLETIM DE CIENCIAS GEODESICAS | 2021年 / 27卷 / 02期
关键词
fine-tuning; pedestrian detection; training data; deep learning models; OBJECT DETECTION;
D O I
10.1590/s1982-21702021000200013
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Object detection in high resolution images is a new challenge that the remote sensing community is facing thanks to introduction of unmanned aerial vehicles and monitoring cameras. One of the interests is to detect and trace persons in the images. Different from general objects, pedestrians can have different poses and are undergoing constant morphological changes while moving, this task needs an intelligent solution. Fine-tuning has woken up great interest among researchers due to its relevance for retraining convolutional networks for many and interesting applications. For object classification, detection, and segmentation fine-tuned models have shown state-of-the-art performance. In the present work, we evaluate the performance of fine-tuned models with a variation of training data by comparing Faster Region-based Convolutional Neural Network (Faster R-CNN) Inception v2, Single Shot MultiBox Detector (SSD) Inception v2, and SSD Mobilenet v2. To achieve the goal, the effect of varying training data on performance metrics such as accuracy, precision, F1-score, and recall are taken into account. After testing the detectors, it was identified that the precision and recall are more sensitive on the variation of the amount of training data. Under five variation of the amount of training data, we observe that the proportion of 60%-80% consistently achieve highly comparable performance, whereas in all variation of training data Faster R-CNN Inception v2 outperforms SSD Inception v2 and SSD Mobilenet v2 in evaluated metrics, but the SSD converges relatively quickly during the training phase. Overall, partitioning 80% of total data for fine-tuning trained models produces efficient detectors even with only 700 data samples.
引用
收藏
页数:16
相关论文
共 50 条
  • [1] Fine tuning deep learning models for breast tumor classification
    Heikal, Abeer
    El-Ghamry, Amir
    Elmougy, Samir
    Rashad, M. Z.
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [2] Knee Implant Identification by Fine-Tuning Deep Learning Models
    Sukkrit Sharma
    Vineet Batta
    Malathy Chidambaranathan
    Prabhakaran Mathialagan
    Gayathri Mani
    M. Kiruthika
    Barun Datta
    Srinath Kamineni
    Guruva Reddy
    Suhas Masilamani
    Sandeep Vijayan
    Derek F. Amanatullah
    Indian Journal of Orthopaedics, 2021, 55 : 1295 - 1305
  • [3] Knee Implant Identification by Fine-Tuning Deep Learning Models
    Sharma, Sukkrit
    Batta, Vineet
    Chidambaranathan, Malathy
    Mathialagan, Prabhakaran
    Mani, Gayathri
    Kiruthika, M.
    Datta, Barun
    Kamineni, Srinath
    Reddy, Guruva
    Masilamani, Suhas
    Vijayan, Sandeep
    Amanatullah, Derek F.
    INDIAN JOURNAL OF ORTHOPAEDICS, 2021, 55 (05) : 1295 - 1305
  • [4] Pedestrian Detection Based on Deep Learning
    Jeon, Hyung-Min
    Vinh Dinh Nguyen
    Jeon, Jae Wook
    45TH ANNUAL CONFERENCE OF THE IEEE INDUSTRIAL ELECTRONICS SOCIETY (IECON 2019), 2019, : 144 - 149
  • [5] Experiments on Fine Tuning Deep Learning Models With News Data For Tweet Classification
    Hallac, Ibrahim R.
    Ay, Betul
    Aydin, Galip
    2018 INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND DATA PROCESSING (IDAP), 2018,
  • [6] Deep Learning Based Pedestrian Detection
    Sun, Weicheng
    Zhu, Songhao
    Ju, Xuewen
    Wang, Dongsheng
    PROCEEDINGS OF THE 30TH CHINESE CONTROL AND DECISION CONFERENCE (2018 CCDC), 2018, : 1007 - 1011
  • [7] Research of deep learning in pedestrian detection
    Zhang, Chuan-Wei
    Yang, Meng-Yue
    Zeng, Hong-Jun
    Li, Bo
    Journal of Computers (Taiwan), 2020, 31 (05) : 249 - 260
  • [8] Joint Deep Learning for Pedestrian Detection
    Ouyang, Wanli
    Wang, Xiaogang
    2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2013, : 2056 - 2063
  • [9] A Deep Transfer Learning Approach to Fine-Tuning Facial Recognition Models
    Luttrell, Joseph
    Zhou, Zhaoxian
    Zhang, Yuanyuan.
    Zhang, Chaoyang
    Gong, Ping
    Yang, Bei
    Li, Runzhi
    PROCEEDINGS OF THE 2018 13TH IEEE CONFERENCE ON INDUSTRIAL ELECTRONICS AND APPLICATIONS (ICIEA 2018), 2018, : 2671 - 2676
  • [10] A comparative study of fine-tuning deep learning models for plant disease identification
    Too, Edna Chebet
    Li Yujian
    Njuki, Sam
    Liu Yingchun
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2019, 161 : 272 - 279