Pytorch
文章平均质量分 56
Mr_寒路
这个作者很懒,什么都没留下…
展开
-
pytorch中转one-hot的两种简单方法
pytorch中转one-hot的两种简单方法1.替换零矩阵arr = torch.tensor([2, 5, 6, 8, 6])zero_arr = torch.zeros(arr.shape[0], max(arr) + 1) #生成零矩阵zero_arr[torch.arange(arr.shape[0]), arr] = 1 #one-hot的位置替换为1print(zero_arr)2.用对角阵index = torch.tensor([2, 5, 6, 8, 6]) # t原创 2021-01-23 19:34:56 · 1740 阅读 · 0 评论 -
Pytorch加载部分权重
Pytorch加载部分权重在图像算法领域,我们有时会遇到训练了很长时间的网络,有时可能需要对原生网络做一些修改。比如在人脸检测项目中,已经训练好人脸框的回归,但是此时需要再加入人脸关键点,修改输出后,此时再加载网络参数一定会报错。为了节约大量时间,我们可以迁移学习加载部分权重。网络权重一般存储在字典中,...原创 2021-01-16 15:29:24 · 1836 阅读 · 0 评论 -
Pytorch中TensorBoard及torchsummary的使用
目录1.TensorBoard1.1 调用方法1.1.1 SummaryWriter1.1.2 add_scalar()1.1.3 add_histogram()1.1.4 add_image()1.1.5 add_graph()2.torchsummary1.TensorBoardTensorBoard是一个强大的可视化工具,在pytorch中有两种调用方法:1.from tensorboardX import SummaryWriter这种方法是在官方还不支持tensorboard时网上有大神写原创 2020-12-20 17:49:15 · 2786 阅读 · 2 评论 -
Pytorch中transforms的多种用法
一、 裁剪——Crop1.随机裁剪:transforms.RandomCrop2.中心裁剪:transforms.CenterCrop3.随机长宽比裁剪 transforms.RandomResizedCrop4.上下左右中心裁剪:transforms.FiveCrop5.上下左右中心裁剪后翻转: transforms.TenCrop二、翻转和旋转——Flip and Rotation6.依概率p水平翻转transforms.RandomHorizontalFlip7.依概率p垂直翻转tra转载 2020-12-03 14:27:44 · 2743 阅读 · 0 评论