需要修改 build_product.yaml
python python/build_gallery.py -c configs/build_product.yaml -o IndexProcess.data_file="D:\Paddle\PaddleClas\out\recognition_demo_data_v1.0\gallery_sg\data_file.txt" -o IndexProcess.index_path="D:\Paddle\PaddleClas\out\recognition_demo_data_v1.0\gallery_sg\index"
python tools/infer.py -c configs/ppyolo/ppyolo_r50vd_dcn_voc.yml --infer_dir=/home/aiuser/mtl/data/sg/test --output_dir=./sg_out_put/ --draw_threshold=0.5 -o weights=./output/ppyolo_r50vd_dcn_voc/model_final
python tools/export_model.py -c configs/ppyolo/ppyolo_r50vd_dcn_voc.yml \
--output_dir=./inference_model \
-o weights=output/ppyolo_r50vd_dcn_voc/model_final
以下列表可以通过--help查看
| FLAG | 支持脚本 | 用途 | 默认值 | 备注 |
|---|---|---|---|---|
| -c | ALL | 指定配置文件 | None | 配置模块说明请参考配置模块 |
| -o | ALL | 设置配置文件里的参数内容 | None | 使用-o配置相较于-c选择的配置文件具有更高的优先级。例如：-o use_gpu=False max_iter=10000 |
| -r/–resume_checkpoint | train | 从某一检查点恢复训练 | None | -r output/faster_rcnn_r50_1x/10000 |
| –eval | train | 是否边训练边测试 | False | |
| –output_eval | train/eval | 编辑评测保存json路径 | 当前路径 | --output_eval ./json_result |
| –fp16 | train | 是否使用混合精度训练模式 | False | 需使用GPU训练 |
| –loss_scale | train | 设置混合精度训练模式中损失值的缩放比例 | 8.0 | 需先开启--fp16后使用 |
| –json_eval | eval | 是否通过已存在的bbox.json或者mask.json进行评估 | False | json文件路径在--output_eval中设置 |
| –output_dir | infer | 输出预测后可视化文件 | ./output | --output_dir output |
| –draw_threshold | infer | 可视化时分数阈值 | 0.5 | --draw_threshold 0.7 |
| –infer_dir | infer | 用于预测的图片文件夹路径 | None | |
| –infer_img | infer | 用于预测的图片路径 | None | 相较于--infer_dir具有更高优先级 |
| –use_vdl | train/infer | 是否使用VisualDL记录数据，进而在VisualDL面板中显示 | False | VisualDL需Python>=3.5 |
| –vdl_log_dir | train/infer | 指定 VisualDL 记录数据的存储路径 | train:vdl_log_dir/scalar infer: vdl_log_dir/image | VisualDL需Python>=3.5 |