fastacvnet++模型嵌入式部署求助

[zky19960]

最近想将fastacvnet++部署到RK3576上，大佬能否指点一下 提个小建议，源代码里面

att_topk = torch.gather(att_weights_prob, 2, ind_k) #源代码

可以换成 #-------------------------------------------------修改部分-----------------------------------------------------# # -------------------------------------------------v0.1-----------------------------------------------------# att_4d = att_weights_prob.squeeze(1) # [B,D,H,W] ind_4d = ind_k.squeeze(1) # [B,k,H,W] att_topk = torch.gather(att_4d, 1, ind_4d) disparity_sample_topk = ind_k.squeeze(1).float() # ------------------------------------------------修改部分-----------------------------------------------------# 因为源代码导出onnx这里的node是5D-tensor，而大部分嵌入式平台对4D-tensor更友好