当前位置: 首页 > 工具软件 > PaddleClas > 使用案例 >

【PaddleClas】常用命令

裴畅
2023-12-01

【PaddleClas】常用命令

1.生成数据集

需要修改 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"

3.预测

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

4.导出

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支持脚本用途默认值备注
-cALL指定配置文件None配置模块说明请参考配置模块
-oALL设置配置文件里的参数内容None使用-o配置相较于-c选择的配置文件具有更高的优先级。例如:-o use_gpu=False max_iter=10000
-r/–resume_checkpointtrain从某一检查点恢复训练None-r output/faster_rcnn_r50_1x/10000
–evaltrain是否边训练边测试False
–output_evaltrain/eval编辑评测保存json路径当前路径--output_eval ./json_result
–fp16train是否使用混合精度训练模式False需使用GPU训练
–loss_scaletrain设置混合精度训练模式中损失值的缩放比例8.0需先开启--fp16后使用
–json_evaleval是否通过已存在的bbox.json或者mask.json进行评估Falsejson文件路径在--output_eval中设置
–output_dirinfer输出预测后可视化文件./output--output_dir output
–draw_thresholdinfer可视化时分数阈值0.5--draw_threshold 0.7
–infer_dirinfer用于预测的图片文件夹路径None
–infer_imginfer用于预测的图片路径None相较于--infer_dir具有更高优先级
–use_vdltrain/infer是否使用VisualDL记录数据,进而在VisualDL面板中显示FalseVisualDL需Python>=3.5
–vdl_log_dirtrain/infer指定 VisualDL 记录数据的存储路径train:vdl_log_dir/scalar infer: vdl_log_dir/imageVisualDL需Python>=3.5
 类似资料: