Pytorch-STN

Pytorch-STN copied to clipboard

Detaching STN input

2

Hello, can I ask you a question? Why [the input of conv1 in SpatialTransformer](https://github.com/aicaffeinelife/Pytorch-STN/blob/master/models/STNModule.py#L49) should be detached? Look forward to your reply. _Originally posted by @wuwenbin970731 in https://github.com/aicaffeinelife/Pytorch-STN/issues/5#issuecomment-659167815_

janzd

Wrong conv layer in the STN localization network

1

Hi! I think that conv4 should be called on https://github.com/aicaffeinelife/Pytorch-STN/blob/master/models/STNModule.py#L54 instead of conv3. You call conv3 before that already? Am I right? I can send a PR if you want.

janzd

The convolutional layers in localization network

4

Hi, thanks for your code. I want to plug the STN into a CNN, and after comparing your code with the code in[ pytorch tutorial](https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html), I find some differences. ...

clelouch

can't reproduce cifar10

13

Can't reproduce cifar10 result. Here is mine without stn: 78.3% with stn: 76.7% Any idea?

twmht