PaddlePaddle
文章平均质量分 73
sapienst
Let Data Speak
展开
-
PaddleDetection使官方使用手册细节点总结(2):模型部署
1、导出可预测模型训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,需要通过tools/export_model.py导出该模型。同时,会导出预测时使用的配置文件,路径与模型保存路径相同, 配置文件名为infer_cfg.yml。输入部分: 导出模型输入为网络输入图像,即原始图片经过预处理后的图像,具体预处理方式可参考配置文件中TestReader部分。各类检测模型的输入格式分别为:输出部分: 导出模型输出统一为NMS的输出,形状为[N, 6], 其中N为预测框原创 2021-03-14 16:35:52 · 1481 阅读 · 0 评论 -
PaddleDetection使官方使用手册细节点总结(1)
1、Fine-tune其他任务使用预训练模型fine-tune其他任务时,可以直接加载预训练模型,形状不匹配的参数将自动忽略,例如:export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7python -u tools/train.py -c configs/faster_rcnn_r50_1x.yml -o pretrain_weights=output/faster_rcnn_r50_1x/model_final也可以显示的指定忽略参数名,可采用如下两种方式:(原创 2021-03-14 15:19:28 · 1057 阅读 · 3 评论