UFOD: An AutoML framework for the construction, comparison, and combination of object detection models

被引:1
作者
Garcia-Dominguez, Manuel [1 ]
Dominguez, Cesar [1 ]
Heras, Jonathan [1 ]
Mata, Eloy [1 ]
Pascual, Vico [1 ]
机构
[1] Univ La Rioja, Dept Math & Comp Sci, Ed CCT C Madre Dios 53, E-26004 Logrono, La Rioja, Spain
关键词
AutoML; Deep learning; Object detection; Transfer learning;
D O I
10.1016/j.patrec.2021.01.022
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Object detection models based on deep learning techniques have been successfully applied in several contexts; however, non-expert users might find challenging the use of these techniques due to several reasons, including the necessity of trying different algorithms implemented in heterogeneous libraries, the configuration of hyperparameters, the lack of support of many state-of-the-art algorithms for training them on custom datasets, or the variety of metrics employed to evaluate detection algorithms. These challenges have been tackled by the development of UFOD, an automated machine learning framework that trains several object detection algorithms (using different underlying frameworks and libraries), compares them, and finally selects the best model or ensembles them. Currently, the most well-known object detection algorithms have been included in our system, and new methods can be easily incorporated thanks to a high-level API. UFOD is available at https://github.com/ManuGar/UFOD/ ? 2021 Elsevier B.V. All rights reserved.
引用
收藏
页码:135 / 140
页数:6
相关论文
共 38 条
  • [1] Abdulla W, 2017, Mask r-cnn for object detection and instance segmentation on keras and tensorflow
  • [2] Agarwal S., 2019, ABS180903193 CORR
  • [3] Ahmad I, 2019, AUTOMATIC DETECTION
  • [4] [Anonymous], 2018, maskrcnn-benchmark: Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch
  • [5] [Anonymous], 2019, XUANNIANZ FSAF FEATU
  • [6] [Anonymous], 2019, XUANNIANZ EFFICIENTD
  • [7] [Anonymous], 2019, XUANNIANZ FCOS FULLY
  • [8] [Anonymous], 2018, MICROSOFT FACEBOOK O
  • [9] [Anonymous], 2017, DEEP LEARNING NOW OP
  • [10] [Anonymous], 2018, DEEP LEARNING MODEL