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鈥檒l occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add PadTo to kornia.augmentation #1286
Conversation
for more information, see https://pre-commit.ci
@shijianjian I did the skeleton - please check about some of the details. Also not sure about how to test the AugmentationSequential. |
Co-authored-by: Jian Shi <sj8716643@126.com>
for more information, see https://pre-commit.ci
def compute_transformation(self, image: torch.Tensor, params: Dict[str, torch.Tensor]) -> torch.Tensor: | ||
return self.identity_matrix(image) | ||
|
||
def apply_transform( | ||
self, image: torch.Tensor, params: Dict[str, torch.Tensor], transform: Optional[torch.Tensor] = None | ||
self, input: torch.Tensor, params: Dict[str, torch.Tensor], transform: Optional[torch.Tensor] = None |
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.
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.
the function signature will need to align with the parent class. If you gonna work on it, I would suggest to do it at once, not for one or two functions for consistency. Also, please keep in mind about the deprecation warnings.
Changes
Fixes #1283
Type of change
Checklist