-
pytorch-mcn:将MatConvNet模型转换为PyTorch的Matlab代码注释规范
资源介绍
matlab代码注释标准pytorch-mcn
用于将经过训练的MatConvNet模型导入Pytorch的工具(如果您希望采用其他方法,请尝试)。
演示版
要运行导入程序,请设置MatConvNet模型的路径,并在importer.sh脚本中提供输出目录(将在其中存储导入的PyTorch模型)。
脚本中有一些示例,可以对其进行注释/取消注释以作为演示运行。
然后运行bash
importer.sh
。
导入模型
已导入并验证了许多标准模型,可以找到。
确认
验证导入的模型需要MATLAB和MatConvNet的副本(具体依赖关系在compare/startup.m中给出)。
过程如下:
运行compare/featureDumper.m脚本以将中间特征从原始MatConvNet模型转储到磁盘。
在debug_mode
(可以在importer.sh设置的选项)中将模型导入PyTorch。这将在PyTorch模型定义中生成其他源代码,该代码存储网络计算的每个中间张量。
运行compare/compare_models.py脚本,该脚本将在张量之间执行数值比较。
笔记
框架之间的模型转换