Skip to content
Snippets Groups Projects
  1. Mar 12, 2021
    • Haian Huang(深度眸)'s avatar
      [Refactor]: Support batch inference traceable by ONNX in SSD, YOLOv3, FSAF,... · 4febf34f
      Haian Huang(深度眸) authored
      [Refactor]: Support batch inference traceable by ONNX in SSD, YOLOv3, FSAF, RetinaNet, and FCOS (#4699)
      
      * Support batch infer in RetinaNet
      
      * Suport batch multiclass_nms
      
      * Revert multiclass_nms
      
      * Fix api deprecated warning
      
      * do not repeat anchors
      
      * Move img_shapes
      
      * Update Yolov3
      
      * Support FCOS
      
      * Support RPN
      
      * Fix RPN topk_inds error
      
      * make batch exportable to onnx for yolohead
      
      * make fcos_head exportable to onnx with batch dim
      
      * Support ATSS
      
      * Support CornerNet and centripetalNet
      
      * Update RetinaNet and delta_xywh
      
      * Remove ugly code
      
      * Remove ugly code of FCOS
      
      * Remove ugly code of ATSS/YOLOV3
      
      * Support RPN and revert bbox_head
      
      * expand anchors to batch and remove BG class when use deploy_nms_pre
      
      * Update
      
      * Use dim=-1 instead of dim=2
      
      * Rename anchor_head method
      
      * Keep the original format output when nms is not use
      
      * Rename method and unified code style
      
      * Fix paa_head and unittest
      
      * Fix FASF onnx export error
      
      * Fix error
      
      * fix single stage img_shapes for onnx
      
      * move conf_thr
      
      * fix rpn_head for onnx
      
      * Add distance2bbox unittest
      
      * Remove TODO
      
      * Fix RPN
      
      * Update docstrs
      
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      Unverified
      4febf34f
    • Haian Huang(深度眸)'s avatar
      Fix RPN show_result error (#4716) · 895a8d51
      Haian Huang(深度眸) authored
      * Fix RPN show_result error
      
      * Remove EmbeddingRPN show_result
      
      * Add docstr
      Unverified
      895a8d51
  2. Mar 09, 2021
  3. Mar 08, 2021
  4. Mar 05, 2021
  5. Mar 03, 2021
  6. Mar 02, 2021
  7. Feb 26, 2021
    • RunningLeon's avatar
      [Feature]: Support onnx2tensorrt for single stage detectors (#4569) · 1085e3c3
      RunningLeon authored
      * Ignore nonzero and add dummy data in multiclass_nms for deployment with TensorRT
      
      * reduce number of bboxes by nms_pre_deploy in anchor_head
      
      * reduce number of bboxes by nms_pre_deploy in fcos_head
      
      * add onnx2tensorrt and its doc
      
      * add nms_pre_deploy in yolo_head
      
      * change sort to topk in rpn_head
      
      * skip nonzero in single_level_roi_extractor
      
      * update onnx2tensorrt and its doc
      
      * move onnx2tensorrt
      
      * update onnx2tensorrt doc
      
      * change nms_pre_deploy to deploy_nms_pre
      
      * update doc and change deploy_nms_pre format
      Unverified
      1085e3c3
    • Cao Yuhang's avatar
      [Feature]: support Feature Pyramid Grids (#4645) · e32ed8d1
      Cao Yuhang authored
      * add cfg
      
      * add cfg
      
      * fix typo
      
      * fix pad
      
      * fix test pad
      
      * add readme
      
      * modify by comments
      
      * fix typo
      Unverified
      e32ed8d1
  8. Feb 25, 2021
  9. Feb 24, 2021
  10. Feb 21, 2021
  11. Feb 19, 2021
    • Jon Crall's avatar
      Dev/mask doctests (#4614) · df681d81
      Jon Crall authored
      
      * Add doctest examples to mask structures
      
      * Add doctests examples to mask_target
      
      * Expanded docs and added examples to loss and get_seg_masks in FCNMaskHead
      
      * Add doctest and update docs of mask_cross_entropy
      
      * Modifications based on review comments
      
      * Fixed boxes to be inside the target image
      
      * Fixed doc in cross entropy loss
      
      * moved from after import
      
      * Added tenative scale_factor and rescale docs
      
      * Moved ensure_rng to top level utils
      
      * Added random classmethods to BitmapBoxes and Polygons
      
      * update docstring of get_seg_masks()
      
      * use height / width instead of H, W
      
      * Better scale_factor rescale docs
      
      * fix linting errors to pass CI
      
      Co-authored-by: default avatarKai Chen <chenkaidev@gmail.com>
      Unverified
      df681d81
  12. Feb 18, 2021
  13. Feb 17, 2021
  14. Feb 09, 2021
  15. Feb 08, 2021
  16. Feb 07, 2021
  17. Feb 05, 2021
  18. Feb 01, 2021
  19. Jan 29, 2021
  20. Jan 28, 2021
    • Thang Vu's avatar
      Code for AAAI 2021 paper. "SCNet: Training Inference Sample Consistency for... · 40f16893
      Thang Vu authored
      Code for AAAI 2021 paper. "SCNet: Training Inference Sample Consistency for Instance Segmentation" (#4356)
      
      * add scnet
      
      * add more configs and minors
      
      * fix unittest for scnet
      
      * update docstring
      
      * add forward test for htc and scnet
      
      * fix build on pytorch 1.3 for empty tensor
      
      * update doc string and minor refactor
      
      * update unit-test and use inheritance for SCNet heads
      
      * support scnet tta
      
      * minor
      
      * update readme
      
      * minor fixes
      Unverified
      40f16893
  21. Jan 27, 2021
  22. Jan 26, 2021
    • Thang Vu's avatar
    • ZhangShilong's avatar
      Reimplement Sparse R-CNN (#4219) · d1edc783
      ZhangShilong authored
      
      * draft version train and test
      
      * remove get_bbox in sparse_head
      
      * refactor SparseRCNN model, align inference eval mAP with official repo
      
      * add focal_loss in hungarian_assigner.py
      
      * update hungarian_assigner.py
      
      * update detr config
      
      * update doc and codes
      
      * fix unitest
      
      * fix unitest
      
      * add embedding_rpn and traing code
      
      * modify code format
      
      * fix proposalist in simpletest
      
      * add focal loss traing
      
      * fix acc
      
      * fix reduce mean
      
      * add embedding rpn doc and add show initproposals
      
      * add docstr
      
      * fix backbone lr
      
      * fix typo and change detach_proposal_list to proposal_list
      
      * remove debug assert
      
      * add readme
      
      * add config for random crop when use 300 proposals
      
      * add aug for 300 proposals
      
      * add 300 proposals results and fix docstr
      
      * add 1x results
      
      * fix lint
      
      * fix unitest
      
      * fix docstr of embeddingrpn
      
      * fix according to the comments
      
      * fix get_bbox and a bug of dynamic conv
      
      * support cpu focal loss and fix docstr
      
      * fix docstr
      
      * fix unitest
      
      * fix bug of  init_weight
      
      * fix to device in assign
      
      * fix all commemt
      
      * fix test_config
      
      * remove dim 1 in forward
      
      * fix shape in comment
      
      * fix the case that train_cfg is None
      
      * fix shape of docstr
      
      * fix docstr
      
      * fix typo
      
      * simplify test
      
      * fix docstr
      
      * move train_cfg and test_cfg to model as
      
      * fix unitest
      
      * add link of ckpt and log
      
      * fix 1x results
      
      Co-authored-by: default avatartianyuandu <tianyuandu@gmail.com>
      Co-authored-by: default avatarfangkairen <fangkairen@sensetime.com>
      Unverified
      d1edc783
Loading