深度学习
文章平均质量分 93
little student
目标检测/模型压缩/红外与可见光图像融合
展开
-
pytorch一行实现:计算同一tensor矩阵内每行之间的余弦相似度
文章目录0 输入数据1 余弦相似度(Cosine Similarity)2 torch.cosine_similarity3 问题4 分析与解决4.1 答案0 输入数据import torch# 设置随机数种子,以保证结果可重现torch.manual_seed(0)a = torch.randn(4, 3)tensor([[ 1.5410, -0.2934, -2.1788], [ 0.5684, -1.0845, -1.3986], [ 0.4033, 0原创 2022-04-09 22:15:34 · 8924 阅读 · 5 评论 -
手把手教你下载ILSVRC2012(ImageNet2012)分类部分数据集以及使用Pytorch训练
ILSVRC2012下载+训练原创 2022-03-24 18:27:32 · 15569 阅读 · 10 评论