Spatial Transformer Network

 STN架構中,分爲三個部分: Localisation net Grid generator Sampler Localisation net  把feature map U作爲輸入,過連續若干層計算(如卷積、FC等),迴歸出參數,在我們的例子中就是一個[2,3]大小的6維仿射變換參數,用於下一步計算; Grid generator  在source圖中找到用於做插值(雙線性插值)的grid。這
相關文章
相關標籤/搜索