Skip to content
Snippets Groups Projects
  • Chrisfsj2051's avatar
    754ec972
    Implement NASFCOS (#2682) · 754ec972
    Chrisfsj2051 authored
    * Implement NAS-FCOS
    
    1. Supports NASFCOS
    2. Added NASFCOS_FPN
    3. Added NASFCOS_Head
    4. Added r50 config
    
    * Revert "Implement NAS-FCOS"
    
    This reverts commit ebae11cbbdc5b4f3cc7226c763e9254c94a9c92b.
    
    * Revert "Revert "Implement NAS-FCOS""
    
    This reverts commit 32bca597859a447de62ddf0f2c2f482809730da9.
    
    * Implement NASFCOS
    
    1. Supports NASFCOS
    2. Added NASFCOS_FPN
    3. Added NASFCOS_Head
    4. Added r50 config
    
    * reformatted & use list to build convs in head
    
    1. code refornatted
    2. add docstring in nasfcos_head.py
    3. use a list of conv config to build convs, rather than names
    
    * cells merged
    
    1. merge cells in nasfpn ans nasfcos_fpn
    2. add nasfpn_cell_factory.py
    
    * reformatted
    
    * fixed
    
    * fixed
    
    * Update nasfpn_cell_factory.py
    
    * pre-commit
    
    * bug fixed
    
    * renamed&moved
    
    * not done
    
    * fixed
    
    * fixed
    
    * Delete remove_parawise_cfg_nasfcos_r50_nashead_4x4_coco.py
    
    * fixed
    
    * Create 1x_bias_lr.py
    
    * Revert "Create 1x_bias_lr.py"
    
    This reverts commit 54602ded283a8f646aebfa0f216350fd2ce82e6d.
    
    * fixed
    
    * fixed
    
    * fixed docstring
    
    * init modified
    
    * formatted
    
    * fixed
    
    * updated
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * pass test
    
    * fixed
    
    * fixed
    
    * fixed
    
    * rename
    
    * 123
    
    * nasfcos_readme
    
    * Update README.md
    
    * rename & leave empty
    
    * GN->GN-head in README
    
    * fixed
    Implement NASFCOS (#2682)
    Chrisfsj2051 authored
    * Implement NAS-FCOS
    
    1. Supports NASFCOS
    2. Added NASFCOS_FPN
    3. Added NASFCOS_Head
    4. Added r50 config
    
    * Revert "Implement NAS-FCOS"
    
    This reverts commit ebae11cbbdc5b4f3cc7226c763e9254c94a9c92b.
    
    * Revert "Revert "Implement NAS-FCOS""
    
    This reverts commit 32bca597859a447de62ddf0f2c2f482809730da9.
    
    * Implement NASFCOS
    
    1. Supports NASFCOS
    2. Added NASFCOS_FPN
    3. Added NASFCOS_Head
    4. Added r50 config
    
    * reformatted & use list to build convs in head
    
    1. code refornatted
    2. add docstring in nasfcos_head.py
    3. use a list of conv config to build convs, rather than names
    
    * cells merged
    
    1. merge cells in nasfpn ans nasfcos_fpn
    2. add nasfpn_cell_factory.py
    
    * reformatted
    
    * fixed
    
    * fixed
    
    * Update nasfpn_cell_factory.py
    
    * pre-commit
    
    * bug fixed
    
    * renamed&moved
    
    * not done
    
    * fixed
    
    * fixed
    
    * Delete remove_parawise_cfg_nasfcos_r50_nashead_4x4_coco.py
    
    * fixed
    
    * Create 1x_bias_lr.py
    
    * Revert "Create 1x_bias_lr.py"
    
    This reverts commit 54602ded283a8f646aebfa0f216350fd2ce82e6d.
    
    * fixed
    
    * fixed
    
    * fixed docstring
    
    * init modified
    
    * formatted
    
    * fixed
    
    * updated
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * fixed
    
    * pass test
    
    * fixed
    
    * fixed
    
    * fixed
    
    * rename
    
    * 123
    
    * nasfcos_readme
    
    * Update README.md
    
    * rename & leave empty
    
    * GN->GN-head in README
    
    * fixed
Code owners
Assign users and groups as approvers for specific file changes. Learn more.

NAS-FCOS: Fast Neural Architecture Search for Object Detection

Introduction

@article{wang2019fcos,
  title={Nas-fcos: Fast neural architecture search for object detection},
  author={Wang, Ning and Gao, Yang and Chen, Hao and Wang, Peng and Tian, Zhi and Shen, Chunhua},
  journal={arXiv preprint arXiv:1906.04423},
  year={2019}
}

Results and Models

Head Backbone Style GN-head Lr schd Mem (GB) Inf time (fps) box AP Download
NAS-FCOSHead R-50 caffe Y 1x 39.4 model | log
FCOSHead R-50 caffe Y 1x 38.5 model | log

Notes:

  • To be consistent with the author's implementation, we use 4 GPUs with 4 images/GPU.