I'm usually against this type of framework baiting, but being a tensorflow guy myself & having just spent the week coding with pytorch full time.... this is basically identical to pytorch
Doing image classification, object localization, and homography (given an input image, which of my known template images is matches it and in what orientation).