forked from pytorch/vision
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[fbsync] Add SimpleCopyPaste augmentation (pytorch#5825)
Summary: * 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 Differential Revision: D37212651 fbshipit-source-id: 467b670164150dd5cc424f4d616d436295ce818d Co-authored-by: vfdev-5 <vfdev.5@gmail.com> Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
- Loading branch information
1 parent
f9f6782
commit b3a6867
Showing
2 changed files
with
177 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters