-
Notifications
You must be signed in to change notification settings - Fork 7.2k
AdjustBrightness/Contrast/Hue/Saturation CV-CUDA Backend #9297
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
AdjustBrightness/Contrast/Hue/Saturation CV-CUDA Backend #9297
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9297
Note: Links to docs will display an error until the docs builds have been completed. ❗ 1 Active SEVsThere are 1 currently active SEVs. If your PR is affected, please view them below: ❌ 4 New Failures, 1 Unrelated FailureAs of commit 13f6e7c with merge base aa35ca1 ( NEW FAILURES - The following jobs have failed:
FLAKY - The following job failed but was likely due to flakiness present on trunk:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
efef692 to
8564f0a
Compare
zy1git
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I left some comments for the first round review. Feel free to let me know your thoughts.
| assert mae < 2 | ||
| mae_threshold = 2 | ||
| if make_input is make_image_cvcuda: | ||
| mae_threshold = 3 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
May I know why the threshold becomes 3 here?
| rgb = cvcuda.cvtcolor(hsv_modified, cvcuda.ColorConversion.HSV2RGB) | ||
|
|
||
| if fp: | ||
| rgb = cvcuda.convertto(rgb, cvcuda.Type.F32, scale=256.0 - 1e-3, offset=0.0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What is the reason for using "256.0 - 1e-3" as the scale?
Summary
Implements the
adjust_brightness,adjust_contrast,adjust_hue, and 'adjust_saturation' CV-CUDA kernelsTesting
python3 -m pytest test/test_transforms_v2.py::TestAdjustBrightnesspython3 -m pytest test/test_transforms_v2.py::TestAdjustContrastpython3 -m pytest test/test_transforms_v2.py::TestAdjustHuepython3 -m pytest test/test_transforms_v2.py::TestAdjustSaturationpython3 -m pytest test/test_transforms_v2.py::TestColorJitter