1. 20 Apr, 2019 2 commits
    • zimenglan's avatar
      add documentation for finetuning cityscapes (#697) · 4466eb5a
      zimenglan authored
      * make pixel indexes 0-based for bounding box in pascal voc dataset
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * add GroupNorm
      
      * add GroupNorm -- sort out yaml files
      
      * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes
      
      * modification on 'group_norm' and 'conv_with_kaiming_uniform' function
      
      * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication
      
      * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead
      
      * agnostic-regression for bbox
      
      * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN
      
      * add README.md for GN
      
      * add dcn from mmdetection
      
      * add documentation for finetuning cityscapes
      
      * add documentation for finetuning cityscapes
      
      * add documentation for finetuning cityscapes
      4466eb5a
    • qianyizhang's avatar
      a44d65dc
  2. 19 Apr, 2019 4 commits
    • Simon Layton's avatar
      Initial mixed-precision training (#196) · 08fcf12f
      Simon Layton authored
      * Initial multi-precision training
      
      Adds fp16 support via apex.amp
      Also switches communication to apex.DistributedDataParallel
      
      * Add Apex install to dockerfile
      
      * Fixes from @fmassa review
      
      Added support to tools/test_net.py
      SOLVER.MIXED_PRECISION -> DTYPE \in {float32, float16}
      apex.amp not installed now raises ImportError
      
      * Remove extraneous apex DDP import
      
      * Move to new amp API
      08fcf12f
    • ChenJoya's avatar
      proposals from RPN per image during training (#676) · bf043792
      ChenJoya authored
      * proposals from RPN per image during training
      
      * README
      
      * Update README for setting FPN_POST_NMS_TOP_N_TRAIN
      
      * Update README.md
      
      * removing extra space change
      bf043792
    • CoinCheung's avatar
      add color jitter augmentation (#680) · 862347d5
      CoinCheung authored
      * add color jitter augmentation
      
      * fix spelling
      862347d5
    • zimenglan's avatar
      add dcn from mmdetection (#693) · 1d6e9add
      zimenglan authored
      * make pixel indexes 0-based for bounding box in pascal voc dataset
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * add GroupNorm
      
      * add GroupNorm -- sort out yaml files
      
      * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes
      
      * modification on 'group_norm' and 'conv_with_kaiming_uniform' function
      
      * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication
      
      * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead
      
      * agnostic-regression for bbox
      
      * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN
      
      * add README.md for GN
      
      * add dcn from mmdetection
      1d6e9add
  3. 16 Apr, 2019 1 commit
  4. 13 Apr, 2019 2 commits
  5. 11 Apr, 2019 1 commit
  6. 10 Apr, 2019 1 commit
  7. 09 Apr, 2019 1 commit
    • Csaba Botos's avatar
      Support Binary Mask with transparent SementationMask interface (#473) · b4d54657
      Csaba Botos authored
      * support RLE and binary mask
      
      * do not convert to numpy
      
      * be consistent with Detectron
      
      * delete wrong comment
      
      * [WIP] add tests for segmentation_mask
      
      * update tests
      
      * minor change
      
      * Refactored segmentation_mask.py
      
      * Add unit test for segmentation_mask.py
      
      * Add RLE support for BinaryMaskList
      
      * PEP8 black formatting
      
      * Minor patch
      
      * Use internal  that handles 0 channels
      
      * Fix polygon slicing
      b4d54657
  8. 05 Apr, 2019 1 commit
  9. 04 Apr, 2019 1 commit
    • Zhang Liliang's avatar
      Fix a bug in Docker file (#637) · 5c44ca74
      Zhang Liliang authored
      Fix a bug.
      Romove the echo command in line 36: RUN conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch 
      To enable conda installation of pytorch-nightly.
      5c44ca74
  10. 02 Apr, 2019 1 commit
  11. 31 Mar, 2019 1 commit
  12. 27 Mar, 2019 1 commit
  13. 26 Mar, 2019 3 commits
    • Miguel Varela Ramos's avatar
      Fixes conda CUDA version in Dockerfile (#604) · bd39d2c1
      Miguel Varela Ramos authored
      * fixes to dockerfile
      
      * replaces local installation by git clone
      bd39d2c1
    • Miguel Varela Ramos's avatar
      Rollback dispatch patch (#603) · 4a7dcc4d
      Miguel Varela Ramos authored
      * Merge branch 'master' of /home/braincreator/projects/maskrcnn-benchmark with conflicts.
      
      * rolls back the breaking AT dispatch changes (#555)
      
      * revert accidental docker changes
      
      * revert accidental docker changes (2)
      4a7dcc4d
    • kaiJIN's avatar
      Support for running on arbitrary CUDA device. (#537) · f0318794
      kaiJIN authored
      * support for any one cuda device
      
      * Revert "support for any one cuda device"
      
      This reverts commit 0197e4e2ef18ec41cc155f3ae2a0face5b77e1e9.
      
      * support runnning for anyone cuda device
      
      * using safe CUDAGuard rather than intrinsic CUDASetDevice
      
      * supplement a header dependency (test passed)
      
      * Support for  arbitrary GPU device.
      
      * Support for arbitrary GPU device.
      
      * add docs for two method to control devices
      f0318794
  14. 25 Mar, 2019 1 commit
  15. 13 Mar, 2019 1 commit
  16. 12 Mar, 2019 2 commits
  17. 11 Mar, 2019 1 commit
  18. 10 Mar, 2019 1 commit
  19. 08 Mar, 2019 2 commits
  20. 05 Mar, 2019 2 commits
  21. 01 Mar, 2019 1 commit
  22. 28 Feb, 2019 2 commits
  23. 26 Feb, 2019 1 commit
    • zimenglan's avatar
      add README.md for GN (#494) · 1818bb2f
      zimenglan authored
      * make pixel indexes 0-based for bounding box in pascal voc dataset
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * add GroupNorm
      
      * add GroupNorm -- sort out yaml files
      
      * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes
      
      * modification on 'group_norm' and 'conv_with_kaiming_uniform' function
      
      * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication
      
      * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead
      
      * agnostic-regression for bbox
      
      * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN
      
      * add README.md for GN
      1818bb2f
  24. 23 Feb, 2019 1 commit
    • Preston Parry's avatar
      Fixes bug with double (overwriting) RESNETS params (#488) · f40f2993
      Preston Parry authored
      There were two `RESNETS` sections, which overrode each other, leading to error messages like: 
      
      ```
      RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN:
      	size mismatch for backbone.fpn.fpn_inner1.weight: copying a param with shape torch.Size([256, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([1024, 256, 1, 1]).
      ...
      	size mismatch for roi_heads.mask.feature_extractor.mask_fcn1.weight: copying a param with shape torch.Size([256, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 1024, 3, 3]).
      ```
      
      This just combines them back into a single section, while maintaining all param values. That got the model running again for me.
      f40f2993
  25. 22 Feb, 2019 1 commit
  26. 20 Feb, 2019 1 commit
    • Stzpz's avatar
      Supported FBNet architecture. (#463) · b23eee0c
      Stzpz authored
      * Supported any feature map size for average pool.
      * Different models may have different feature map size.
      
      * Used registry to register keypoint and mask heads.
      
      * Passing in/out channels between modules when creating the model.
      
      Passing in/out channels between modules when creating the model. This simplifies the code to compute the input channels for feature extractors and makes the predictors independent of the backbone architectures.
      * Passed in_channels to rpn and head builders.
      * Set out_channels to model modules including backbone and feature extractors.
      * Moved cfg.MODEL.BACKBONE.OUT_CHANNELS to cfg.MODEL.RESNETS.BACKBONE_OUT_CHANNELS as it is not used by all architectures. Updated config files accordingly.
      
      For new architecture modules, the return module needs to contain a field called 'out_channels' to indicate the output channel size.
      
      * Added unit test for box_coder and nms.
      
      * Added FBNet architecture.
      
      * FBNet is a general architecture definition to support efficient architecture search and MaskRCNN2GO.
      * Included various efficient building blocks (inverted residual, shuffle, separate dw conv, dw upsampling etc.)
      * Supported building backbone, rpn, detection, keypoint and mask heads using efficient building blocks.
      * Architecture could be defined in `fbnet_modeldef.py` or in `cfg.MODEL.FBNET.ARCH_DEF` directly.
      * A few baseline architectures are included.
      
      * Added various unit tests.
      
      * build and run backbones.
      * build and run feature extractors.
      * build and run predictors.
      
      * Added a unit test to verify all config files are loadable.
      b23eee0c
  27. 19 Feb, 2019 3 commits
    • zimenglan's avatar
      please set 'STRIDE_IN_1X1' to be 'False' when backbone uses GN (#465) · 192261db
      zimenglan authored
      * make pixel indexes 0-based for bounding box in pascal voc dataset
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * replacing all instances of torch.distributed.deprecated with torch.distributed
      
      * add GroupNorm
      
      * add GroupNorm -- sort out yaml files
      
      * use torch.nn.GroupNorm instead, replace 'use_gn' with 'conv_block' and use 'BaseStem'&'Bottleneck' to simply codes
      
      * modification on 'group_norm' and 'conv_with_kaiming_uniform' function
      
      * modification on yaml files in configs/gn_baselines/ and reduce the amount of indentation and code duplication
      
      * use 'kaiming_uniform' to initialize resnet, disable gn after fc layer, and add dilation into ResNetHead
      
      * agnostic-regression for bbox
      
      * please set 'STRIDE_IN_1X1' to be 'False' when backbone use GN
      192261db
    • xelmirage's avatar
      modify test() to run_test() (#460) · f8b01183
      xelmirage authored
      I'm using pycharm to debug the code on a remote server, the remote debugging seems to be performed by pytest and it pops errors like:
      train_net.py E
      test setup failed
      file /tmp/pycharm_project_269/tools/train_net.py, line 79
        def test(cfg, model, distributed):
      E       fixture 'cfg' not found
      >       available fixtures: cache, capfd, capfdbinary, caplog, capsys, capsysbinary, doctest_namespace, monkeypatch, pytestconfig, record_property, record_xml_attribute, recwarn, tmp_path, tmp_path_factory, tmpdir, tmpdir_factory
      >       use 'pytest --fixtures [testpath]' for help on them.
      
      
      it seems the function name ‘test()’ has come conflict with pytest, so it may be better use another name.
      f8b01183
    • Csaba Botos's avatar
      remove note for detectron dependency (#458) · 5ec0b91c
      Csaba Botos authored
      5ec0b91c