• Vasilis Vryniotis's avatar
    Post-paper Detection Optimizations (#5444) · 08cc9a7f
    Vasilis Vryniotis authored
    * Use frozen BN only if pre-trained.
    
    * Add LSJ and ability to from scratch training.
    
    * Fixing formatter
    
    * Adding `--opt` and `--norm-weight-decay` support in Detection.
    
    * Fix error message
    
    * Make ScaleJitter proportional.
    
    * Adding more norm layers in split_normalization_params.
    
    * Add FixedSizeCrop
    
    * Temporary fix for fill values on PIL
    
    * Fix the bug on fill.
    
    * Add RandomShortestSize.
    
    * Skip resize when an augmentation method is used.
    
    * multiscale in [480, 800]
    
    * Add missing star
    
    * Add new RetinaNet variant.
    
    * Add tests.
    
    * Update expected file for old retina
    
    * Fixing tests
    
    * Add FrozenBN to retinav2
    
    * Fix network initialization issues
    
    * Adding BN support in MaskRCNNHeads and FPN
    
    * Adding support of FasterRCNNHeads
    
    * Introduce norm_layers in backbone utils.
    
    * Bigger RPN head + 2x rcnn v2 models.
    
    * Adding gIoU support to retinanet
    
    * Fix assert
    
    * Add back nesterov momentum
    
    * Rename and extend `FastRCNNConvFCHead` to support arbitrary FCs
    
    * Fix linter
    08cc9a7f
test_models.py 32.3 KB