
Pytorch
文章平均质量分 65
不变强不改名
求关注
展开
-
numpy中的赋值与copy()
numpy中的赋值与copy()原创 2022-05-04 09:56:01 · 948 阅读 · 0 评论 -
pytorch中tensor的直接赋值与clone()、numpy()
关于pytorch中tensor中莫名其妙的改变或输出莫名其妙原创 2022-05-03 16:57:14 · 7711 阅读 · 0 评论 -
qt模板用于展示cv算法(使用vs studio+qt插件实现)
使用qt show出你的cv算法原创 2022-04-26 18:48:41 · 1418 阅读 · 0 评论 -
VS2017中NCNN使用vulkan
VS2017中NCNN使用vulkan原创 2022-04-25 23:46:45 · 4263 阅读 · 0 评论 -
深度学习中常见的损失函数——关于使用与prediction与label的shape说明
计算机视觉中常用的损失函数,理解prediction与label的shape以及损失计算方式原创 2022-04-25 23:37:52 · 5323 阅读 · 0 评论 -
WSL2调用Windows中的GPU版Pytorch
在wsl2使用GPU训练pytorch模型原创 2022-04-04 15:31:58 · 2275 阅读 · 0 评论 -
模型压缩(4)——知识蒸馏
知识蒸馏,高效压缩模型的简单实现原创 2022-03-19 10:43:25 · 1430 阅读 · 0 评论 -
模型压缩(3)——模型量化
模型量化,高效压缩模型的简单实现原创 2022-03-18 10:27:08 · 1934 阅读 · 0 评论 -
pytorch快手上手——从读取数据到ncnn部署
pytorch入门指导,从读取数据到ncnn部署,全流程原创 2022-03-18 09:29:59 · 3572 阅读 · 0 评论 -
模型压缩(2)——模型剪枝
模型剪枝,高效压缩模型的简单实现原创 2022-03-12 23:40:15 · 5763 阅读 · 1 评论 -
模型压缩(1)——低秩分解
低秩分解,高效压缩模型的简单实现原创 2022-03-09 19:16:02 · 3241 阅读 · 0 评论 -
超最简单的安卓端深度学习模型部署教程——pytorch2onnx2pdinfer2android
超简单的实现方式,快把你的深度学习模型安装到小伙伴的android手机吧!!!原创 2021-12-12 18:15:16 · 7518 阅读 · 6 评论 -
pytorch预测结果的精度、召回率、F1值、准确率计算(简单实现)
1. 导入必要的库函数import osimport jsonimport torchfrom PIL import Imagefrom torchvision import transforms# 导入自己的模型from model_v3 import mobilenet_v3_small2. 文件存储格式imgs文件下下是需要预测的图像label.txt为每幅图对应的标签### 这个无所谓,能正常读取就行3. 标签读取辅助函数targets_path =.原创 2021-11-22 21:31:59 · 13184 阅读 · 4 评论 -
Pytorch冻结预训练权重(特征提取与BN层)
1. 读取预训练权重pre_weights = torch.load(model_weights_path, map_location=device)2. 读取预训练权重中与现有模型参数设置相同层的权重,可适用于修改了分类或某些层通道数的情况net = yourmodel()pre_dict = {k: v for k, v in pre_weights.items() if net.state_dict()[k].numel() == v.numel()}# strict原创 2021-11-22 09:59:40 · 3360 阅读 · 2 评论