Skip to content
Snippets Groups Projects
  1. Aug 05, 2020
    • robin Han's avatar
      Pytorch2onnx (#3075) · 2f32a47a
      robin Han authored
      
      * Update pytorch2onnx.py which using new logic to convert pytorch to ONNX
      
      * use standard API to check whether in ONNX convert process
      
      * only compare the score value while verifying results between ONNX and pytorch
      
      * move import onnx before import torch, or something weird will happen
      
      * use real images for input
      
      * modifying the way of calling nms
      
      * modify docstring for bbox2result, and remove unnecessary part for onnx exporting
      
      * modify the 'Convert to ONNX' part in docs
      
      * replace or to | in docstring
      
      * update according to the latest mmcv
      
      * add normalize part
      
      * raise error while using low version mmcv
      
      * minor update
      
      * minor update
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      Unverified
      2f32a47a
  2. Jul 04, 2020
  3. Jun 30, 2020
    • implus's avatar
      GFocal (#3097) · 39e6257c
      implus authored
      * add gfocal losses
      
      * add gfl head, its related code, configs
      
      * fix 2x config bug
      
      * add most model and log in gfl README
      
      * update gfl model and log
      
      * Update transforms.py
      
      modify variables  L, T, R, B   to  left, top, right, bottom
      
      * Update gfocal_loss.py
      
      modify variables into meaningful names:
      * pt -> scale_factor
      * a -> pos   b -> pos_label
      * disl -> dis_left  disr -> dis_right
      * wl -> weight_left  wr -> weight_right
      
      * Update gfocal_loss.py
      
      update formats
      
      * remove duplicated get_bboxes/IoU calculation, make QFL consistent
      
      * add all docstring in gfl_head.py and gfocal_loss.py
      Unverified
      39e6257c
  4. Jun 26, 2020
  5. Jun 07, 2020
  6. May 31, 2020
  7. Apr 19, 2020
  8. Apr 08, 2020
    • Wenwei Zhang's avatar
      Refactor category orders in heads (#2374) · 70eedbd3
      Wenwei Zhang authored
      
      * Refactor (all): all category -1 in anchor/bbox head and anchor/bbox target assign
      
      * Fix (datasets): remove label + 1 in datasets
      
      * Fix (bbox_head): fix bug of fc_cls that forget + 1
      
      * Fix (atss_head & free_anchor): fix cat -1 bugs
      
      * Fix (mask_head): remove label + 1 in mask heads
      
      * fix atss
      
      * Fix (rpn): fix cross_entropy_loss bug of RPN
      
      * Fix (anchor_head): fix typo
      
      * Refactor (anchor_head): use background_label rather than num_classes to indicate background class
      
      * Refactor (docstring): add and reformat docstrings
      
      * fix mask iou head
      
      * Fix (mask_head): fix cat -1 bug
      
      * Fix (mask_head): fix bug in mask inference
      
      * Add (tests): add tests for mask rcnn and mask heads
      
      * Refactor (unittest): refactor test_forward
      
      * Refactor (new_empty): use new_full rather than new_empty
      
      * Refactor (background_label): check background_label
      
      * Add TODO
      
      * Refactor (unittest): allow BP in unittest
      
      Co-authored-by: default avatarCao Yuhang <yhcao6@gmail.com>
      Unverified
      70eedbd3
  9. Apr 06, 2020
    • Wenwei Zhang's avatar
      Change V2.0 coors (#2380) · 7ed8d51e
      Wenwei Zhang authored
      * Refactor (all): change coordinate system
      
      * Fix (mask_head): fix cat -1 bug in mask_paste
      
      * Fix (unittest)
      : modify unittest and pass CI
      
      * reformat to pass CI
      
      * Fix round coordinates bugs
      
      * clean file
      
      * Fix (test): use cpu version of aligned roi_align in tests
      
      * Refactor (mask): clean np.stack
      
      * Refactor (head): reformat code and fix missing -1
      
      * Reformat: reformat and add doc strings
      
      * Refactor (mask_head): more clea docstring
      Unverified
      7ed8d51e
  10. Mar 21, 2020
  11. Oct 07, 2019
    • Jon Crall's avatar
      Doctests for AnchorHead and related code (#1491) · e344e21d
      Jon Crall authored
      * Doctests for AnchorHead and related code
      
      * Fix flake8 error
      
      * resolve review on transforms.py
      
      * Resolve comments
      
      * resolve commend about foreground classes
      
      * Add reference about bounding box delta encoding
      e344e21d
  12. May 17, 2019
    • Cao Yuhang's avatar
      Reimplement "FCOS: Fully Convolutional One-Stage Object Detection" (#586) · 8d38fd8c
      Cao Yuhang authored
      * add fcos
      
      * use P5 instead of C5
      
      * add relu before extra convs in FPN
      
      * add singleclass_nms, use caffe2 lr
      
      * fix log interval
      
      * use caffe2init and relu in extra layers
      
      * fix scale layer, use p5 instead of c5
      
      * fix focs target
      
      * refactor code
      
      * delete useless file
      
      * clean
      
      * refactor code
      
      * change num_classes to cls_out_channels
      
      * fix bug of in get_bboxes
      
      * fix bug in test
      
      * add r101 2x cfg
      
      * ms use value mode, add x101-64x4d cfg
      
      * add more comment and rename some variable
      
      * rename centers to points, modify doc string of distance2bbox
      
      * add fcos detector, replace frozen with requires_grad
      
      * add README.md
      
      * add r101-1x performance, rename cfg, add detector FCOS
      
      * update fcos r50 2x performance, remove fpn caffe2 initialize
      
      * fix flake8 error
      
      * rename cfg
      
      * fix grammar error of some comments
      
      * minor fix comment
      
      * change work_dir to be consistent with config name
      
      * add FCOS support in README
      8d38fd8c
  13. Oct 08, 2018
  14. Oct 05, 2018
  15. Sep 23, 2018
  16. Sep 12, 2018
  17. Sep 02, 2018
Loading