【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	支持脚本	用途	默认值	备注
-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