bubbliiiing / unet-pytorch

这是一个unet-pytorch的源码,可以训练自己的模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问可以写一个inference_onnx.py么

Vegeta975 opened this issue · comments

这个detect_image部分处理方式是对输入图像进行网络的前向传递计算得到预测值存入pr中,如果不引用unet的结构,只提供unet.onnx模型,这部分该如何读取呢

我加个待办,会弄一个,之前在目标检测弄了一个

话说之前加了你有看着吗0 0