![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
刀么克瑟拉莫
那温热的牛奶瓶在你手中握紧
展开
-
Monodepth2和Lite-Mono准备数据集
【代码】Monodepth2和Lite-Mono准备数据集。原创 2023-08-15 15:01:36 · 484 阅读 · 2 评论 -
pytorch加载部分权重的方法
【代码】pytorch加载部分权重的方法。原创 2023-06-13 14:38:13 · 217 阅读 · 0 评论 -
使用log_softmax而不是softmax
log_softmax计算过程原创 2023-04-10 17:00:37 · 680 阅读 · 0 评论 -
pytorch实现加权交叉熵
pytorch 加权交叉熵原创 2023-03-29 10:44:06 · 695 阅读 · 0 评论 -
yolox之SimOTA代码解析与详细注释
yolox SimOTA原创 2023-02-17 15:05:04 · 574 阅读 · 0 评论 -
pytorch取每行给定位置的数字
pytorch取每行给定位置的数字gather原创 2022-09-02 10:36:36 · 712 阅读 · 0 评论 -
pytorch保存一张图的所有特征图
pytorch 特征图原创 2022-08-30 11:54:53 · 957 阅读 · 6 评论 -
pytorch权重文件pth转pt或onnx
pth pt onnx原创 2022-08-16 15:54:01 · 1663 阅读 · 0 评论 -
极简版pytorch实现yolov3-tiny
极简版 pytorch yolov3-tiny原创 2022-08-12 17:14:37 · 3185 阅读 · 4 评论 -
pytorch加载darknet权重文件
pytorch加载darknet权重文件原创 2022-08-10 10:49:12 · 2954 阅读 · 0 评论 -
pytorch的技巧记录
大于某数的索引原创 2022-07-26 20:08:09 · 175 阅读 · 0 评论 -
pytorch之repeat尽可能全方位的理解
pytorch之repeat原创 2022-07-21 10:16:17 · 368 阅读 · 0 评论 -
yolov3后处理,包括网络输出、阈值过滤、多类NMS
grid_h, grid_w, IOU, NMS原创 2022-06-15 16:52:35 · 1243 阅读 · 1 评论 -
pytorch统计模型的参数个数
就是遍历每一个参数,把大小加起来import torchdevice = torch.device('cuda' if torch.cuda.is_available() else 'cpu')paras=torch.load('weights.pth', map_location=device)count = 0for k2 in paras: l = list(paras[k2].size()) temporary = 1 for i in l: temporary *= i.原创 2022-05-31 10:38:03 · 410 阅读 · 0 评论 -
pytorch相同结构不同参数名模型加载权重
因为声明的网络模型比保存的模型bn层多一个参数num_batches_tracked,所以跳过它import torch, copynet=yourNet()state_dict = copy.deepcopy(net.state_dict())keys=[]for key in state_dict: if(key.split('.')[-1] == 'num_batches_tracked'): continue keys.append(key)device = torch.d.原创 2022-05-31 10:33:44 · 760 阅读 · 1 评论 -
pytorch中的repeat
在yolov3中,要将anchor的宽度[[116],[156],[373]]复制成batch_size*3*scaled_w*scaled_h的形状,故对repeat的方式做了一下研究。取batch_size=2,scaled_w=scaled_h=5import torchw=torch.Tensor([[116],[156],[373]])u=w.repeat(50,1).view(2,3,5,5)v=w.repeat(1,50).view(2,3,5,5)w=w.repeat(2,.原创 2022-02-10 18:43:07 · 830 阅读 · 1 评论 -
pytorch简洁实现MNIST识别
可以看这里,比较与tensorflow版本的区别在输入格式方面,pytorch是NCHW,tensorflow是NHWC网络返回log_softmax时,应该使用nll_lossMyData.pyimport osimport cv2import randomimport numpy as npclass Dataset(object): def __init__(self, dataset_path, train, batch_size=1): self.al.原创 2022-01-04 17:16:38 · 1060 阅读 · 0 评论