pytorch
bearmomo
奋进的菜鸟,专注图像处理~
展开
-
pytorch 打印自定义模型并正确寻找每一层的名字
再我们生成热力图或者需要找到自定义模型的某一层时,这时候就需要通过某一层的名字来定位该层。当我们自定义模型后,那么寻找该层的名字时,就通过观察模型输出的架构。通过print(Net)得到下面的模型。那么我想找到其中某层的名字,就看带有:的前()中的名字。凡是():这种形式的都是模型的层的名称序列。通过缩进看名称的包含关系。比如我使用layer_name='pretrained_model.layer1.0.group1.conv3'找到其中某一层卷积层。Net( (pretr..原创 2020-06-29 11:20:45 · 4123 阅读 · 0 评论 -
Linux系统使用pyinstaller打包pytorch模型疯狂踩坑记(含权重文件和图片等)
整了一天半,整到心态爆炸,终于整好了,哭了。首先,先自学了pyqt5编写界面,编写的过程就下次再说吧。编写完之后,文件夹中包括权重文件和两张图片。作为小白,咱也不知道这个打包竟然不能打包图片和权重文件,百度了好几圈终于知道了,/(ㄒoㄒ)/~~然后这些图片咋整啊,后来看到了大佬写的代码,把图片先转成py文件,再从py文件中解码出来。代码如下:1.图片转py文件代码:import base64def pic2py(picture_name): """ 将图像文件转换原创 2020-05-26 19:15:52 · 2444 阅读 · 5 评论