Skip to content

Commit

Permalink
comment new transforms
Browse files Browse the repository at this point in the history
  • Loading branch information
themattinthehatt committed Nov 1, 2024
1 parent 6ab33ea commit 714d28e
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions lightning_pose/data/augmentations.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,9 @@ def imgaug_transform(cfg: DictConfig) -> iaa.Sequential:
iaa.CoarseDropout(p=prct, size_percent=size_prct, per_channel=per_channel)
))
# coarse salt and pepper
# bright reflections can often confuse the model into thinking they are paws
# (which can also just be bright blobs) - so include some additional transforms that put
# bright blobs (and dark blobs) into the image
prct = 0.01 # probability of changing a pixel to salt/pepper noise
size_prct = (0.05, 0.1) # drop pix on a low-res version of img that's `size_prct` of og
data_transform.append(iaa.Sometimes(
Expand Down

0 comments on commit 714d28e

Please sign in to comment.