苞米面Paddle助手是一个百度飞桨Paddle、PaddleX项目模板和小工具。
适用系统一些脚本使用shell编写,所以目前适用Linux和百度AIStudio如何安装从gitee获取源码gitclonegit@gitee.com:cnhemiya/bmm-paddle-helper.git从github获取源码gitclonegit@github.com:cnhemiya/bmm-paddle-helper.git程序参数项目生成器mkbmmph.pycdbmm-paddle-helperpython3tools/mkbmmph.py-husage:mkbmmph.py[-h][--project][--to_dir]苞米面Paddle项目生成器optionalarguments:-h,--helpshowthishelpmessageandexit--project项目类型,可选:paddlex_cls,paddlex_det,paddlex_seg--to_dir生成的项目模板保存目录,默认'./run'生成不同的模板cdbmm-paddle-helper#PaddleX图像分类python3tools/mkbmmph.py--projectpaddlex_cls--to_dir./run#PaddleX目标检测python3tools/mkbmmph.py--projectpaddlex_det--to_dir./run#PaddleX图像分割python3tools/mkbmmph.py--projectpaddlex_seg--to_dir./runLinux小技巧可以使用软连接把mkbmmph.py连接到$HOME/.local/bin目录,方便使用。
使用示例生成模板生成PaddleX目标检测项目模板cdbmm-paddle-helperpython3tools/mkbmmph.py--projectpaddlex_det--to_dir./run模板目录结构run├──aismain.ipynb├──check_data.sh├──get_data.sh├──infer.py├──mod│├──args.py│├──config.py│├──__init__.py│├──pdxconfig.py│├──report.py│└──utils.py├──onekey.sh├──onetasks.sh├──paddlex_det_doc.md├──prune.py├──quant.py└──train.py文件说明文件说明aismain.ipynbJupyternotebook适用百度AIStudio,放到项目根目录,或者根据目录修改check_data.sh检查数据get_data.sh获取数据infer.py预测程序modpython模块onekey.sh一键获取数据脚本模板,需要按照自己需求,修改路径onetasks.sh一键训练,量化脚本模板,需要按照自己需求,修改参数paddlex_det_doc.md参数说明prune.py裁剪程序quant.py量化程序train.py训练程序aismain.ipynb目录结构示例├──aismain.ipynb├──data└──work└──run训练示例train.py加-h查看参数python3run/train.py\\--dataset./dataset/road_fighter_car\\--epochs32\\--batch_size1\\--learning_rate0.01\\--modelPicoDet\\--backboneESNet_m\\--pretrain_weights""裁剪示例prune.py加-h查看参数python3run/prune.py\\--dataset./dataset/road_fighter_car\\--epochs16\\--batch_size1\\--learning_rate0.001\\--model_dir./output/best_model\\--save_dir./output/prune\\--pruned_flops0.2在线量化示例quant.py加-h查看参数python3run/quant.py\\--dataset./dataset/road_fighter_car\\--epochs16\\--batch_size1\\--learning_rate0.001\\--model_dir./output/best_model\\--save_dir./output/quant需要修改什么aismain.ipynb放到百度AIStudio项目根目录,或者根据目录修改check_data.shdataset_dir:需要检查的文件所在的目录data_files:需要检查的文件onekey.shapp_dir:程序目录,如果不是run,根据自己的设定修改zip_file:数据压缩包ais_dir:aistudio数据目录sub_data_dir:解压后的数据目录data_dir:数据目录onetasks.shMODEL:模型名称BACKBONE:主干模型DATASET:数据集目录BASE_SAVE_DIR:保存的目录FIXED_INPUT_SHAPE:导出模型的输入大小APP_DIR:程序目录PYTHON_APP:python程序transforms可以自己修改train.pyprune.pyquant.py的transforms非必须。dataset可以自己修改train.pyprune.pyquant.py的dataset非必须。开源协议MulanPSL-2.0
项目地址GITEE https://gitee.com/cnhemiya/bmm-paddle-helper
GITHUB https://github.com/cnhemiya/bmm-paddle-helper
AISTUDIO https://aistudio.baidu.com/aistudio/projectdetail/4260357
[点击并拖拽以移动]
评论