2021年02月_jingbo1801

12月 11月 10月 09月 08月 07月 06月 05月 04月 03月 02月 01月

原创 onnxsim-让导出的onnx模型更精简

用torch导出的模型有时候参数过多，不利于查看。查找资料onnxsim可以简化模型，让显示更加自然。如，reshap层的导出，红框中细节参数被显示出现。默认导出：使用onnxsim 可以让结构更加简洁，具体执行方式如下：step1、安装onnxsim包pip install onnx-simplifierstep2、加载onnx文件，simplify处理后重新保存，代码如下：from onnxsim import simplifyonnx_model = onnx.loa

2021-02-22 21:33:08 37559 9

原创 Netron-模型结构可视化工具（支持tf, caffe, keras,mxnet,onnx,darknet）

Netron支持主流各种框架的模型结构可视化工作，github地址：https://github.com/lutzroeder/netronNetron 支持的框架和对应文件如下：ONNX (.onnx, .pb, .pbtxt), Keras (.h5, .keras), CoreML (.mlmodel), Caffe2 (predict_net.pb, predict_net.pbtxt), MXNet (.model, -symbol.json)TensorFlow L

2021-02-22 21:23:34 2983 1

翻译 mmsegment工具包（十）

目录1、获取FLOP和参数2、发布模型3、转换为ONNX4、打印整个配置除了训练/测试脚本，在tools/目录下提供了许多有用的工具。1、获取FLOP和参数提供了一个适用于flops-counter.pytorch的脚本，用于计算给定模型的FLOP和参数。python tools/get_flops.py ${CONFIG_FILE} [--shape ${INPUT_SHAPE}]将得到这样的结果：==============================.

2021-02-03 21:14:58 819

原创 mmsegment训练技巧（九）

目录1、训练技巧1.1、Backbone和Heads不同学习率1.2、在线难样本挖掘（OHEM）1.3、类平衡损失2、自定义优化设置2.1、Pytorch支持的自定义优化器2.2、定制优化器2.3、在配置文件中指定优化器2.4、自定义优化器构造函数2.5、其他设置3、定制训练计划4、自定义钩子过程4.1、使用MMCV中实现的挂钩4.2、修改默认的运行时挂钩4.3、检查点配置4.4、日志配置4.5、评估配置1、训练技巧1.1、Back

2021-02-03 21:09:42 4022 5

原创 mmsegment定制模型（八）

1、定制优化器1.1、优化器注册类mmsegmentation中未对优化器进行封装，真实的封装及注册类在mmcv中完成。我们可以在schedule默认配置中看到优化器的字典型配置。在mmseg/api/train.py中，由build_optimizer完成构造。进一步代码跟进，在mmcv中实现该注册类。所以，不用纠结，工程代码中为什么字典型优化器配置，如何转为pytorch的优化器类。1.2、优化器定义假设定义一个名为优化MyOptimizer，其中有参数a，b和c

2021-02-02 23:33:11 1926 1

原创 mmsegment数据pipeline操作（七）

目录1、数据项配置2、voc数据集传入参数3、CustomDataset数据读取4、self.pipeline(results)4.1、读图4.2、数据增广4.3、格式转换4.4、测试5、扩展和使用自定义管道1、数据项配置不使用数据增广，返回原始默认图像及标签：from mmseg.datasets import build_datasettrain_cfg = dict( type='PascalVOCDataset', data_ro

2021-02-01 22:29:32 4904 2

原创 mmsegment数据集说明（六）

目录1、数据集加载2、加载图片及标记3、定义自己的数据集4、数据集操作4.1、重复数据集4.2、连接数据集https://mmsegmentation.readthedocs.io/en/latest/tutorials/customize_datasets.html1、数据集加载依据tools/train.py中调用方式，提取单独的数据集加载逻辑。from mmcv.utils import Configfrom mmseg.datasets import...

2021-02-01 22:18:15 1932