-
Notifications
You must be signed in to change notification settings - Fork 7.2k
AdjustSharpness CV-CUDA Backend #9301
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?
AdjustSharpness CV-CUDA Backend #9301
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9301
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: ❌ 5 New FailuresAs of commit eee2040 with merge base aa35ca1 ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
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.
| ) -> "cvcuda.Tensor": | ||
| cvcuda = _import_cvcuda() | ||
|
|
||
| if len(_max_value_map) == 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.
Using if not _max_value_map:
| if len(_max_value_map) == 0: | ||
| _max_value_map[cvcuda.Type.U8] = 255 | ||
| _max_value_map[cvcuda.Type.F32] = 1.0 | ||
| if len(_dtype_to_format) == 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.
Using if not _dtype_to_format:
Summary
Adds the CV-CUDA backend kernel for
adjust_sharpnessTesting
python3 -m pytest test/test_transforms_v2.py::TestAdjustSharpness