Skip to content
Snippets Groups Projects
  1. Mar 15, 2021
  2. 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
  3. Mar 09, 2021
  4. Mar 08, 2021
  5. Mar 05, 2021
  6. Mar 03, 2021
  7. Mar 02, 2021
  8. 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
  9. Feb 25, 2021
  10. Feb 24, 2021
  11. Feb 21, 2021
  12. 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
  13. Feb 18, 2021
  14. Feb 17, 2021
  15. Feb 09, 2021
  16. Feb 08, 2021
  17. Feb 07, 2021
  18. Feb 05, 2021
  19. Feb 01, 2021
  20. Jan 29, 2021
  21. 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
  22. Jan 27, 2021
  23. Jan 26, 2021
Loading