Open lmolhw5252 opened 6 years ago
Are these output sizes same?
Yes, the output of the spatial transformer sampling mechanism always has the same spatial size. This could be 50x50 or 100x32, depends on what you want to do.
The sampled size is the same in both images you posted here.
@Bartzi And I got another qusetion,when I train network with STN, the network is difficult to converge.Do you have any trick about training STN? I use image with 1 word.
Yes, there are some tricks you can try:
1e-4
to 1e-6
--load-localization
), this will help the trainingHope this helps ;)
Thanks a lot about your reply!
Hi, I got a problem with STN. I see that STN's params need target dim as 50.Is that means the outputs of STN are feature maps size of [batch_size, channels, output_height, output_width], and both height and width are fixed? What about different size of words in an image? I have not fully understood.