stn.pytorch
stn.pytorch copied to clipboard
fxia22
pytorch version of spatial transformer networks
Hi, Thanks for this awesome project! I have a straightforward question. Why is there a lr parameter in AffineGridGenFunction? https://github.com/Cadene/stn.pytorch/blob/master/script/functions/gridgen.py#L8 Also, is it useful to have a lr parameter in...
Hi, fxia22, 1. I used the 'BCHW' version of STN for affine transformation (to rotate an image), however, the output image does not have the same size as input. Instead,...
I recently tried to use the new BCHW functions with my network, since i always use that layout and it simplifies my code. I noticed that all the gradients of...
Thank you for this implementation! Have you tried using it within a network that is wrapped in a DataParallel in order to make use of multiple graphics cards? I am...
BCHW format
Excellent work! I would like to use this in the middle of my pytorch network, so my tensors are in [Batch x Channel x Height x Width] format. I tried...
Hi, Thank you for sharing great code. I found that the code works fine with gpu_id = 0, but it raises cuda runtime error if I tried to run it...
