基本流程 pytorch模型->onnx模型->（onnx模型简化）->tensorrt模型 pytorch模型导出为onnx模型 dummy_input = torch.randn(50, 3, 1280, 1280, device='cuda') # 导出时推理输入 input_names = ["images"] # 输入名 output_names = ['num_dets', 'det_boxes']
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
