/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/TensorRT/lib:$LD_LIBRARY_PATH
cd SemanticSegmentation
获取预训练...推断文档[13] 给出了模型输入输出:
输入: [src, r1i, r2i, r3i, r4i, downsample_ratio]
src:输入帧,RGB 通道,形状为 [B, C, H, W],范围为...downsample_ratio 必须是 FP32,其他输入必须和加载的模型使用一样的 dtype
输出: [fgr, pha, r1o, r2o, r3o, r4o]
fgr, pha:前景和透明度通道输出,范围为...= gs.import_onnx(onnx.load(input))
_print_graph(graph)
# update node Resize_3: scales...= [i for i in graph.inputs if i.name !