Problem with traffic signs

chenhsuanlin / inverse-compositional-STN

Inverse Compositional Spatial Transformer Networks :performing_arts: (CVPR 2017 oral)

MIT License

318 stars 63 forks source link

Problem with traffic signs #15

Closed DaddyWesker closed 5 years ago

DaddyWesker commented 5 years ago

Hello. Just tried to run your code. Mnist-tensorflow works fine, but traffic signs shows some strange results... Can you tell me, if just running train.py with option, for example IC-STN, is enough in case of signs?

problem_with_signs

here is what i've got.

chenhsuanlin commented 5 years ago

It seems that I set the default opt.lrGP to be too high for IC-STN. I tried running with the learning rate set to be 3e-5 and it seems to be able to train through 1M iterations. This has been updated, sorry about this!