大神,请教下,用自己数据集用官方的weight权重训练完成后,剪枝了,还能正常转成pb吗? about yolov3-channel-and-layer-pruning

[tanluren]

这个项目的权重可以选择保存为pt, darknet weights, 或者onnx格式，搜索一下很多方法转为pb，没验证过就不具体推荐了，但要注意的是，剪枝的模型结构发生了改变，要保证转换的tf模型和新的cfg具有相同的结构。tf剪枝的话，官方有模型优化工具包，但貌似并不实际‘剪’，你可以参考这个项目移植到tf里面，原理和方法并不难，可以看看各剪枝脚本的代码，特别是prune_utils的细节处理；要考虑两个基本的问题，一是什么样的规则确定哪些通道或者层是重要或者不重要的，二是怎么拿到保留的位置的权重。想快速动手写代码的话，也可以GitHub搜一下tensorflow pruning，参考相关开源项目是怎么写的。