Skip to content
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

SpatialTransformer #233

Merged
merged 2 commits into from
Jul 22, 2020
Merged

Conversation

KazukiYoshiyama-sony
Copy link
Contributor

Spatial Transformer is composed of two functions:

  • AffineGrid (2d/3d) generates the normalized grid by the affine transformation.
  • WarpByGrid (2d/3d, linear/nearest, channel_first/last, zero/repeat/reflect pad) warps the input image by the grid generated by AffineGrid.

[CUDA] WarpByGrid

[CUDNN] AffineGrid

[CUDNN] WarpByGrid

Fix flag

Format of WarpByGrid

Fix call function

Fix arguments

Fallback when channel_last or windows in cudnn

[Cuda] use std::array and delete redundant template spec.

[cuda] format

[Cuda] use std::vector again

[Cuda] use std::vector again
@TE-StephenTiedemann TE-StephenTiedemann merged commit dd3b8d2 into master Jul 22, 2020
@KazukiYoshiyama-sony KazukiYoshiyama-sony added the release-note-op-layer Auto-release; function improvemenet or/and addition label Aug 4, 2020
@YukioOobuchi YukioOobuchi deleted the feature/20200616-spatial-transformer branch November 2, 2020 06:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
release-note-op-layer Auto-release; function improvemenet or/and addition
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants