分类：pth文件

2021
03-05

从Pytorch模型pth文件中读取参数成numpy矩阵的操作

目的：把训练好的pth模型参数提取出来，然后用其他方式部署到边缘设备。Pytorch给了很方便的读取参数接口：nn.Module.parameters()直接看demo：fromtorchvision.models.alexnetimportalexnetmodel=alexnet(pretrained=True).eval().cuda()parameters=model.parameters()forpinparameters:numpy_para=p.detach().cpu().numpy()print(type(numpy_para))print(numpy_para.shape)上面得到的numpy_para就是numpy参数... 继续阅读 >

2020
09-29

使用pth文件添加Python环境变量方式

有时，在用import导入项目文件夹里自己写的python文件时，常常说找不到该文件，这时可以将项目路径添加到PYTHONPATH下。暂时添加：importsyssys.path.append(项目路径)这种方法只会存在内存里，一旦退出python就没有了一劳永逸：在D:/anaconda/env/pytorch/lib/site-packages/里新建txt，修改后缀为.pth，在其中加入你的项目路径，比如K:\Desktop\face_detect\DFace-win64-master\srcK:\Desktop\face_detect\DFace-win64-master\s... 继续阅读 >