pytorch 与 tensorrt upsample对应关系
pytorch:F.upsample(x, size=(H, W), mode='bilinear',align_corners=True)
对应trt:
tensorrt:
IResizeLayer* P5_resize = network->addResize(*P5->getOutput(0));
auto C4_shape = network->addShape(*C4->getOutput(0))->getOutput(0);
P5_resi
原创
2021-04-08 15:10:27 ·
1486 阅读 ·
2 评论