Wrong conv layer in the STN localization network
janzd opened this issue · 1 comments
janzd commented
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.
wuwenbin970731 commented
Hello, can I ask you a question?
Why the input of conv1 in SpatialTransformer should be detached?
Look forward to your reply.