Skip to content
Snippets Groups Projects
  1. May 02, 2020
    • Cao Yuhang's avatar
      Add configs and benchmarks. (#2446) · 482fcfa9
      Cao Yuhang authored
      
      * add configs for all models
      
      * fix some configs
      
      * Add benchmark models
      
      * update some benchmarks
      
      * Fix retina config
      
      * add more benchmarks
      
      * update MODEL_ZOO.md
      
      * fix cascade mask num_class bug
      
      * change caffe pretrained url
      
      * fix paramwise_cfg bug
      
      * fix paramwise cfg
      
      * restrict minimum h and w in crop_and_resize
      
      * add grad clip for fcos
      
      * add grad clip for ga retinanet
      
      * fix grad clip
      
      * fix img norm
      
      * fix rpn loss bbox
      
      * fix rpn proposal num
      
      * fix url
      
      * clean
      
      * update comparison
      
      * minor update
      
      * add grad clip for guided anchoring
      
      * fix some typos
      
      * add inference time details
      
      * clean
      
      Co-authored-by: default avatarZwwWayne <wayne.zw@outlook.com>
      Unverified
      482fcfa9
  2. Apr 11, 2020
  3. Mar 31, 2020
    • Wenwei Zhang's avatar
      Encapsulate the second stage into RoI heads (#1999) · c29a789c
      Wenwei Zhang authored
      * create roi_head
      
      * create roi_head
      mv shared_head into roi_head
      
      * fix conflict
      
      * refactor(cascade rcnn): create cascade head in roi head, simplify cascade rcnn
      
      * type: fix
      scope: cascade_head
      subject: mv stage_loss_weight from train_cfg to roi_head creation
      
      * type: fix
      scope: roi_heads
      subject: handle the case when train_cfg is None
      
      * type: fix
      scope: roi_heads mask inference
      subject: fix test_cfg.rcnn bug to test_cfg
      
      * fix mask eval bug
      
      * refactor(double_head): refactor double head roi head
      
      * fix(double_head): clean code and fix __init__.py
      
      * refactor(mask scoring): create mask scoring roi head
      
      * refactor(htc_head): add htc_head
      
      * fix(htc_head): add htc_head in __init__.py
      
      * fix (htc_head): fix htc_head bugs
      
      * fix (ms_roi_head): rm test_cfg.rcnn to test_cfg
      
      * feature (grid rcnn): grid roi head
      
      * fix (grid_head): fix grid head bug
      
      * reformat and change all cfgs
      
      * reformat (roi_head): reformat __init__.py for isort=4.3.21
      
      * fix roi_head test bug
      
      * fix (carafe): change carafe cfgs to use roi_head
      
      * fix (roi_head): fix missing roi_head refactor
      
      * reformat to pass CI
      
      * test all cfgs
      
      * match keys of configs with previous commit 77d073a
      
      * add and pass unittest for all roi heads
      
      * Refactor (roi_head): extract abstract base class for roi head
      
      * Refactor (roi_head): refactor init functions
      
      * Refactor (roi_head): weight init
      
      * Refactor (roi_head): add _bbox_forwrd & _mask_forward as basic functions
      
      * Fix (grid_roi_head): fix bug in bbox_forward_train
      
      * Refactor (roi_head): change to use img.device in forward_dummy)
      
      * Refactor (roi_heads): simplify init functions and _mask_forward
      
      * Fix (cascade_roi_head): fix init bug of cascade roi_head
      
      * Refactor (roi_head): use dict as outputs of _bbox_forward and _mask_forward
      
      * Refactor (test_config): scan valid configs rather than list them all
      Unverified
      c29a789c
  4. Mar 24, 2020
  5. Mar 23, 2020
    • Jerry Jiarui XU's avatar
      add new config sample and check scripts (#2216) · eeff0bef
      Jerry Jiarui XU authored
      * add new config sample and check scripts
      
      * context managere to prevent import
      
      * refactor backbone, add caffe
      
      * add fast rcnn
      
      * all basci config done
      
      * rm coco_proposal
      
      * add scratch, wider_face, voc
      
      * add reppoints
      
      * delete device id, auto resume
      
      * add libra_rcnn, ms_rcnn, nas_fpn
      
      * add hrnet, fcos
      
      * add free_anchor, gn, grid_rcnn, guided_anchoring, gcnet, ghm, gn+ws
      
      * finished all, but some need refactor and unified logic
      
      * fixed albu
      
      * update ReseNet101
      
      * reduce ResNet101, add norm_cfg, 4conv1fc
      
      * refactor caffe backbone
      
      * merge act_cfg, dcn from master
      
      * rename 2fc head, add 4conv1fc head
      
      * delete 4con1fc model
      
      * rename components -> _base_
      
      * flatten _base_
      
      * refactor _base_ structure
      
      * refactor config structure
      
      * reduce _base/models
      
      * rename all config
      
      * add _coco inside configs
      
      * fixed _coco in files
      
      * fixed albu in test
      
      * migrate to new sys
      
      * fixed test
      
      * fixed test
      
      * rm cascade, fast_rcnn
      
      * rm cascade, fast_rcnn from _base_
      
      * fixed fcos_center
      
      * rename from rule
      
      * collopse third party model
      
      * fixed test
      
      * fixed some minor bug
      
      * rename gn-neck_head -> gn-neck+head
      Unverified
      eeff0bef
Loading