
pytorch
文章平均质量分 71
define_mine
我很懒
展开
-
onnx 模型修改(删除指定层,并改输出层)
删完了之后应该还要把输出层指出来,onnx的输出可以作为节点被删除,但是创建需要走一下他的创建输出流程,输出维度放在了output_shape_map中,自行修改输出维度后,校验一下模型,没问题就可以进行下一步了。名字列表放到我这里定义的x 里面,后面会按从大到小降序删除,如果反着删的话,列表排序的位置会变,就会删错层,这里需要注意。我的诉求是删了红框的部分,开一个 https://netron.app/,看onnx层要删除的层的名字。比如这个sigmoid层,点击右边的name就是名字。原创 2023-06-08 11:12:43 · 2778 阅读 · 0 评论 -
YoloV5 工程文档 学习自记
目录个人记录YOLOv5 工程文件中的需要注意的文件python 类相关定义python 的作用域和命名空间关于self 的代码关于类的继承关于super 函数描述python 基础知识asserta =3 if True else 5and 、or数组切片操作[::2]/[1::2]math.gcd() 求最大公约数nn 相关的几个函数nn.Conv2d语句翻译Conv个人记录正在扒代码,随手记录,上传方便以后找YOLOv5 工程文件中的需要注意的文件runs: 存储结果/中间过程文件weig原创 2021-08-13 23:07:13 · 936 阅读 · 0 评论