Skip to content
Snippets Groups Projects
  1. Jul 07, 2020
    • Wenwei Zhang's avatar
      Add Colab Training Tutorial (#3213) · e2b642a4
      Wenwei Zhang authored
      * Add colab training tutorial
      
      * Update tutorial codes and fix bug
      
      * Clean files, refactor TBD
      
      * Update tutorials
      
      * Remove duplicated code
      
      * Mv tutorial
      e2b642a4
  2. Jul 04, 2020
  3. Jun 26, 2020
  4. Jun 19, 2020
  5. Jun 08, 2020
  6. May 31, 2020
  7. May 27, 2020
  8. May 19, 2020
  9. May 07, 2020
  10. May 05, 2020
  11. May 04, 2020
  12. Apr 29, 2020
    • David de la Iglesia Castro's avatar
      Add option to save the result images of running tools/test (#2414) · e8117f0f
      David de la Iglesia Castro authored
      
      * Use apis/inference in detectors show_result
      
      * Add images_out_dir arg to test apis
      
      * Include --images_out_dir in assertion
      
      * Fix single class parsing
      
      * Remove single class hack
      
      * Move import
      
      * Add example of saving results
      
      * Fix list number
      
      * Refactor show_result
      
      * Update docs
      
      * Update __init__
      
      * Fix CLASSES reference
      
      * Remove unnecessar assert for custom classes
      
      * Use - instead of _
      
      * Use - instead of _
      
      * Rename images_out_dir -> out_dir
      
      * Remove unnecessary model.module
      
      * Update show_result with all params and docstring
      
      * Fix missing comma
      
      * Set rescale always to True
      
      * Remove outdated restriction from docs
      
      * Drop pathlib
      
      * Rename out_dir -> show_dir
      
      * Update docstring
      
      * More explicit code. Fix out_dir
      
      * Update docstrings
      
      * Flake8
      
      Co-authored-by: default avatarmmeendez8 <miguelmndez@gmail.com>
      Co-authored-by: default avatarsbugallo <sbugallo@gradiant.org>
      e8117f0f
  13. Apr 25, 2020
  14. Apr 11, 2020
  15. 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
      c29a789c
  16. Mar 27, 2020
  17. Mar 21, 2020
  18. Mar 13, 2020
  19. Jan 31, 2020
    • Kai Chen's avatar
      Add logger utils (#2035) · 93bed07b
      Kai Chen authored
      * add logger utils
      
      * replace get_root_logger() and logger.info() with print_log()
      
      * fix a typo
      
      * minor fix for the format of StreamHandler
      93bed07b
  20. Dec 31, 2019
  21. Dec 25, 2019
  22. Dec 21, 2019
    • Rinat Shigapov's avatar
      Async inference interface (#1647) · 9729ca54
      Rinat Shigapov authored
      * async inference support
      
      * implemented concurrent decorator
      
      * fixes for Python versions < 3.7
      
      * async methods depend on python version
      
      * revert changes in forward method
      
      * async_test -> async_simple_test, debug logging is done via logger.debug
      
      * add async test
      
      * add asynctest to test requirements
      
      * async tests are run in Python 3.7
      
      * check CUDA, add docs
      
      * fix device
      
      * run test only if CUDA is available
      
      * fix linting
      
      * custom operators can run on nondefault stream
      
      * set current stream in kernel launch configuration
      
      * example fixes
      
      * add async/sync interface comparison benchmark
      
      * fix linting
      9729ca54
  23. Nov 10, 2019
    • Jon Crall's avatar
      Pass img_meta as list to simple_test (#1517) · 4d84161f
      Jon Crall authored
      * Fix simple_test for base and fast_rcnn detector
      
      * Add test of detector forward methods
      
      * Revert forward_test change and add docs indicating expected inputs for forward
      
      * fix linter
      4d84161f
  24. Oct 09, 2019
  25. Aug 23, 2019
    • Kai Chen's avatar
      Make data pre-processing pipeline customizable (#935) · 0d5233a3
      Kai Chen authored
      * define data pipelines
      
      * update two config files
      
      * minor fix for config files
      
      * allow img_scale to be optional and update config
      
      * add some docstrings
      
      * add extra aug to transform
      
      * bug fix for mask resizing
      
      * fix cropping
      
      * add faster rcnn example
      
      * fix imports
      
      * fix robustness testing
      
      * add img_norm_cfg to img_meta
      
      * fix the inference api with the new data pipeline
      
      * fix proposal loading
      
      * delete args of DefaultFormatBundle
      
      * add more configs
      
      * update configs
      
      * bug fix
      
      * add a brief doc
      
      * update gt_labels in RandomCrop
      
      * fix key error for new apis
      
      * bug fix for masks of crowd bboxes
      
      * add argument data_root
      
      * minor fix
      
      * update new hrnet configs
      
      * update docs
      
      * rename MultiscaleFlipAug to MultiScaleFlipAug
      
      * add __repr__ for all transforms
      
      * move DATA_PIPELINE.md to docs/
      
      * fix image url
      0d5233a3
  26. Jul 31, 2019
  27. Jun 28, 2019
    • Cao Yuhang's avatar
      Support FP16 training (#520) · e7466395
      Cao Yuhang authored
      * add fp16 support
      
      * fpn donot need bn normalize
      
      * refactor wrapped bn
      
      * fix bug of retinanet
      
      * add fp16 ssd300 voc, cascade r50, cascade mask r50
      
      * fix bug in cascade rcnn testing
      
      * add support to fix bn training
      
      * add fix bn cfg
      
      * delete fixbn cfg, mv fixbn fp16 to a new branch
      
      * fix cascade mask fp16 bug in test
      
      * fix bug in cascade mask rcnn fp16 test
      
      * add more fp16 cfgs
      
      * add fp16 fast-r50 and faster-dconv-r50
      
      * add fp16 test, minor fix
      
      * clean code
      
      * fix config work_dir name
      
      * add patch func, refactor code
      
      * fix format
      
      * clean code
      
      * move convert rois to single_level_extractor
      
      * fix bug for cascade mask, the seg mask is ndarray
      
      * refactor code, add two decorator force_fp32 and auto_fp16
      
      * add fp16_enable attribute
      
      * add more comment, fix format and test assertion
      
      * fix pep8 format error
      
      * format commont and api
      
      * rename distribute to distributed, fix dict copy
      
      * rename function name
      
      * move function, add comment
      
      * remove unused parameter
      
      * mv decorators into decorators.py, hook related functions to hook
      
      * add auto_fp16 to forward of semantic head
      
      * add auto_fp16 to all heads and fpn
      
      * add docstrings and minor bug fix
      
      * simple refactoring
      
      * bug fix for patching forward method
      
      * roi extractor in fp32 mode
      
      * fix flake8 error
      
      * fix ci error
      
      * add fp16 support to ga head
      
      * remove parallel test assert
      
      * minor fix
      
      * add comment in build_optimizer
      
      * fix typo in comment
      
      * fix typo enable --> enabling
      
      * udpate README
      e7466395
  28. May 17, 2019
  29. Apr 17, 2019
    • myownskyW7's avatar
      Support models without FPN (#133) · a9e21cf7
      myownskyW7 authored
      * add two stage w/o neck and w/ upperneck
      
      * add rpn r50 c4
      
      * update c4 configs
      
      * fix
      
      * config update
      
      * update config
      
      * minor update
      
      * mask rcnn support c4 train and test
      
      * lr fix
      
      * cascade support upper_neck
      
      * add cascade c4 config
      
      * update config
      
      * update
      
      * update res_layer to new interface
      
      * refactoring
      
      * c4 configs update
      
      * refactoring
      
      * update rpn_c4 config
      
      * rename upper_neck as shared_head
      
      * update
      
      * update configs
      
      * update
      
      * update c4 configs
      
      * update according to commits
      
      * update
      a9e21cf7
  30. Dec 11, 2018
  31. Dec 05, 2018
  32. Oct 10, 2018
  33. Oct 05, 2018
  34. Oct 03, 2018
  35. Sep 26, 2018
  36. Sep 23, 2018
Loading