search

aicaffeinelife / Pytorch-STN

Spatial Transformer Networks in Pytorch.
MIT License
155 stars 35 forks source link

Wrong conv layer in the STN localization network #5

Open janzd opened 4 years ago

janzd commented 4 years ago

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 4 years ago

Hello, can I ask you a question? Why the input of conv1 in SpatialTransformer should be detached? Look forward to your reply.