![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 76
资料加载中
迎着风奔跑
展开
-
通过中心点生成heatmap
使用2D高斯函数来构建学习目标(heatmap)。将某一关键点的ground-truth作为中心点,这样一来,中心点处将具有最高的得分,越远离中心点,得分将越低。公式地表示,则有将关键点的grounnd-truth转换为mask,可以使用分割的方式预测k张feature-map对应k个关键点,最后在每个热力图中选择最大值对应的坐标,即为model预测输出的结果。对应代码段如下所示: 参考github:hourglass代码:...原创 2021-08-10 11:23:37 · 1129 阅读 · 0 评论 -
注意力机制的原理及实现(pytorch)
注意力机制大礼包原创 2023-03-14 22:23:22 · 7064 阅读 · 1 评论 -
再探pytorch的Dataset和DataLoader
一探到底Dataset和DataLoade原创 2023-03-17 16:36:29 · 1958 阅读 · 1 评论 -
基于pytorch实现猫狗分类系统
自己实现一个猫狗分类系统原创 2023-03-24 10:46:12 · 2926 阅读 · 20 评论 -
【pytorch源码剖析系列】梯度优化算法的概述和实践
看穿优化器原创 2023-03-24 14:04:02 · 657 阅读 · 0 评论 -
【pytorch源码剖析系列】模型搭建
pytoch模型构建方法原创 2023-03-19 18:38:37 · 278 阅读 · 2 评论 -
pytorch将文件夹中的所有图片变换为一个张量,并使用matplotlib显示
函数功能:实现多张图片拼接为张量,并进行显示函数使用:只需要改变几个超参数img_file_path、height、width、channels.注意:因为将多个多张图片拼接为一个张量,所以所有的图片的shape要相同,否则会报错import matplotlib.pyplot as pltimport torchimport osimport cv2def count_file(img_file_path): img_file_list = os.listdir(img_file_原创 2021-01-14 17:14:28 · 1314 阅读 · 1 评论 -
pytorch自定义数据集的写法
1.自定义数据集继承于基类Dataset。2.该类需要实现三个函数,分别为__init__()、len()、getitem();第一个函数是初始化函数,初始化图片存储路径(列表的形式),第二个函数是获得图片的长度,第三个函数通过index(index是通过dataloader函数获取)加载路径中的图片,进行增强化处理,并获得该图片的标签,最后返回增强处理后的图片和标签。3.transform的包装,通过compose()进行串行序列化包装,对图片进行顺序处理。4.实例化自定义数据集类。class T原创 2020-11-10 09:59:54 · 487 阅读 · 0 评论