首页 PaddleX 帖子详情
目标检测---04---目标检测模型预测及导出
收藏
快速回复
PaddleX 文章入门 905 4
目标检测---04---目标检测模型预测及导出
收藏
快速回复
PaddleX 文章入门 905 4

模型预测代码如下:

model = pdx.load_model('output/fasterrcnn_resnet/best_model')  # 加载训练好的模型
image_name = 'data/JPEGImages/50.jpg'
result = model.predict(image_name)
pdx.det.visualize(image_name, result, threshold=0.6, save_dir='./output/')  # 将预测好的图片标记并输出到指定目录,threhold代表置信度低于0.6的不进行输出

模型转换

    在服务端部署模型时需要将训练过程中保存的模型导出为inference格式模型,因此,需要将训练好的best_model进行导出,在终端执行如下命令如下:

paddlex --export_inference --model_dir=output/fasterrcnn_resnet/best_model/ --save_dir=inference_model/resnet

inference模型如何部,下一篇细讲

0
收藏
回复
全部评论(4)
时间顺序
余志良
#2 回复于2021-08

非常赞

0
回复
李长安
#3 回复于2021-12

优秀

0
回复
DeepGeGe
#4 回复于2021-12

昨天我也做了目标检测模型的部署,不过是使用PaddleDetection套件的现成的工具来使用的。

0
回复
JavaRoom
#5 回复于2021-12

部署后面文档有啊

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户