bevdet-tensorrt-cpp icon indicating copy to clipboard operation
bevdet-tensorrt-cpp copied to clipboard

Published 20 hours ago • verified publisher icon LCH1238

export中尺寸问题

Open zmmhz opened this issue 1 year ago • 0 comments

img_input = torch.zeros([6B, 3, 900, 400], dtype=torch.int32, device=f'cuda:{args.gpu_id}') 请问在转onnx时为什么要设置成900400，不直接设置为900*1600？

zmmhz avatar Dec 14 '23 09:12 zmmhz