Implementation of GaussianMixtureMasksTransform

[snehilchatterjee]

Modified Components:

• lightly/transforms/__init__.py

New Files:

• lightly/transforms/gaussian_mixture_masks_transform.py
• tests/transforms/test_gaussian_mixture_masks.py

Enhancement:

• Added support for Gaussian Mixture Mask transform

Fixes Issue:

• Closes Add GaussianMixtureMask transform #1661

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 84.85%. Comparing base (b8613dc) to head (b68b473).
Report is 1 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #1692      +/-   ##
==========================================
+ Coverage   84.78%   84.85%   +0.07%     
==========================================
  Files         154      155       +1     
  Lines        6479     6509      +30     
==========================================
+ Hits         5493     5523      +30     
  Misses        986      986              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[SauravMaheshkar]

@snehilchatterjee, could you provide some input-output pairs?

[snehilchatterjee]

[guarin]

Thanks a lot for the PR, the augmentations look great! I left some comments to make the interface identical to the other FFT transforms.

[guarin]

Thanks! Reviewed again

[snehilchatterjee]

After making the changes this is the output:

[guarin]

This looks great, thanks so much!

[snehilchatterjee]

Glad I could contribute