Pytorch-STN icon indicating copy to clipboard operation
Pytorch-STN copied to clipboard

Published 20 hours ago • verified publisher icon aicaffeinelife

Wrong conv layer in the STN localization network

Open janzd opened this issue 4 years ago • 1 comments

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 avatar Jul 07 '20 19:07 janzd

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

wuwenbin970731 avatar Jul 16 '20 05:07 wuwenbin970731