• Lezwon Castelino's avatar
    Add SimpleCopyPaste augmentation (#5825) · bbc1aac8
    Lezwon Castelino authored
    
    
    * added simple POC
    
    * added jitter and crop options
    
    * added references
    
    * moved simplecopypaste to detection module
    
    * working POC for simple copy paste in detection
    
    * added comments
    
    * remove transforms from class
    updated the labels
    added gaussian blur
    
    * removed loop for mask calculation
    
    * replaced Gaussian blur with functional api
    
    * added inplace operations
    
    * added changes to accept tuples instead of tensors
    
    * - make copy paste functional
    - make only one copy of batch and target
    
    * add inplace support within copy paste functional
    
    * Updated code for copy-paste transform
    
    * Fixed code formatting
    
    * [skip ci] removed manual thresholding
    
    * Replaced cropping by resizing data to paste
    
    * Removed inplace arg (as useless) and put a check on iscrowd target
    
    * code-formatting
    
    * Updated copypaste op to make it torch scriptable
    Added fallbacks to support LSJ
    
    * Fixed flake8
    
    * Updates according to the review
    Co-authored-by: default avatarvfdev-5 <vfdev.5@gmail.com>
    Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
    bbc1aac8
transforms.py 22.8 KB