bmm-paddle-helper 苞米面 Paddle 助手开源项目

我要开发同款
匿名用户2022年06月23日
46阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议Mulan

作品详情

苞米面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

[点击并拖拽以移动]

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论