当前位置: 首页 > 工具软件 > Kornia > 使用案例 >

pytorch半精度训练使用kornia库中的warp_perspective出现nan值

商昆琦
2023-12-01

把插值mode换成nearest就好了

x = k.geometry.warp_perspective(x, self.pro_mats[cam], self.reduced_worldgrid_shape,mode="nearest")
 类似资料: