unet-pytorch icon indicating copy to clipboard operation
unet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

训练自己的数据集问题

Open PengPeng-JunJun opened this issue 2 years ago • 3 comments

博主你好，已经可以正常训练并进行预测，用的Vgg，在转成ONNX后，在cpu上推理预测，速度很慢，输入图像是30722048，shape是512512，推理时间在2.2S左右，请问这是代码问题还是模型问题呢？

PengPeng-JunJun avatar Feb 28 '23 13:02 PengPeng-JunJun

模型呀，本来就很慢，这么大

bubbliiiing avatar Feb 28 '23 14:02 bubbliiiing

之前训练过一个enet的模型，模型文件的大小大概是1M多一些。请问你有用过基于pytorch框架的比较小的模型做语义分割么？

PengPeng-JunJun avatar Feb 28 '23 14:02 PengPeng-JunJun

还有一个问题请教，之前有用你的yolo4做目标检测，模型文件好像也是200M，还没有转ONNX测试，请问这个的FPS大概是多少呢？在cpu上推理预测，cpu是i7-7700

PengPeng-JunJun avatar Feb 28 '23 15:02 PengPeng-JunJun