a small bug in warp.py when opt.warpType = affine

chenhsuanlin / inverse-compositional-STN

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

MIT License

318 stars 63 forks source link

Closed Coldmooon closed 6 years ago

Coldmooon commented 6 years ago

If the transformation type is affine, there should be 6 parameters. Returning p7 and p8 from torch.unbind(p) will prompt the error.

chenhsuanlin commented 6 years ago

Thanks @Coldmooon for the fix!