Skip to content
Snippets Groups Projects
  1. Mar 03, 2021
  2. Mar 02, 2021
  3. 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
  4. Feb 25, 2021
  5. Feb 24, 2021
  6. Feb 21, 2021
  7. 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
  8. Feb 18, 2021
  9. Feb 17, 2021
  10. Feb 09, 2021
  11. Feb 08, 2021
  12. Feb 07, 2021
  13. Feb 05, 2021
  14. Feb 01, 2021
  15. Jan 29, 2021
  16. 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
  17. Jan 27, 2021
  18. 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
  19. Jan 24, 2021
  20. Jan 20, 2021
    • Haian Huang(深度眸)'s avatar
      [Feature] Support to visualize results based on prediction quality (#4441) · 4310fed1
      Haian Huang(深度眸) authored
      * New Add ResultEvaluate
      
      * Update ResultVisualizer
      
      * Delete debug code
      
      * Add Loading related pipeline
      
      * 1.Move retrieve_loading_pipeline to dataset.util
      2.Support display GT mask
      3.Support display GT mask and det mask for visualize fn
      
      * Use gt_mask.to_ndarray()
      
      * Refactor analyze_results.py
      
      * Fix docs str
      
      * Logger is silent and add progressBar
      
      * Fix lint
      
      * Fix docs str and other comment
      
      * Fix docs str
      
      * Fix docs str
      
      * Fix docs str and add unit test for mask2ndarray
      Unverified
      4310fed1
  21. Jan 19, 2021
  22. Jan 18, 2021
  23. Jan 13, 2021
Loading