Skip to content

Commit

Permalink
Fix a comment
Browse files Browse the repository at this point in the history
Signed-off-by: Kamil Tokarski <[email protected]>
  • Loading branch information
stiepan committed Mar 8, 2023
1 parent b8cd469 commit da572bc
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions dali/python/nvidia/dali/auto_aug/augmentations.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,9 +121,6 @@ def contrast_mean_centered(sample, parameter):
"""
This variant follows PIL implementation of Contrast enhancement, which does not use
the middle of the `dtype` range as the contrast center, but a channel-weighted mean.
NOTE: Currently, it is not possible to run this variant with `sample` that resides
in GPU memory due to reduction values residing in GPU and the limitation that named arguments
(i.e. contrast_center) must reside in CPU memory.
"""
mean = fn.reductions.mean(sample, axes=[0, 1])
rgb_weights = types.Constant(np.array([0.299, 0.587, 0.114], dtype=np.float32))
Expand Down

0 comments on commit da572bc

Please sign in to comment.