模型如何移植到jetson Orin开发板上？

[lkt0105]

纯新手，大佬我想把这个模型移植到英伟达的jetson Orin上转化为TensorRT Engine，使用c++来进行推理实现图像的超分辨率重建。但是现在对于DAT-light模型的input输入形式有些疑惑，我看你定义的img_size=64，这是表明导入图片的输入是3 * 64 * 64吗？我是否需要对图片进行切片为64 * 64之后再进行推理然后再组装成图。

[zhengchen1999]

不需要，img_size=64是为了方便训练设置的参数，可以提前计算w-sa使用到的mask，从而避免重复计算。但是在推理时，任意大小分辨率的输入都可以（重新计算mask即可）。