花费 12 ms
pytorch tensor与numpy转换

从官网拷贝过来的,就是做个学习记录。版本 0.4 tensor to numpy 输出 进行转换 输出 注意,转换 ...

2018-10-01 05:48 1 67497
COCO 数据集的使用

1 简介 微软发布的 COCO 数据库是一个大型图像数据集, 专为对象检测、分割、人体关键点检测、语义分割和字幕生成而设计。 COCO 数据库的网址是: MS COCO 数据集主页:htt ...

2018-04-29 04:07 1 81755
pytorch之dataloader深入剖析

PyTorch学习笔记(6)——DataLoader源代码剖析 - dataloader本质是一个可迭代对象,使用iter()访问,不能使用next()访问; - 使用iter(dataloader)返回的是一个迭代器,然后可以使用next访问; - 也可以使 ...

2018-12-17 05:03 3 52621
Pytorch里的CrossEntropyLoss详解

在使用Pytorch时经常碰见这些函数cross_entropy,CrossEntropyLoss, log_softmax, softmax。看得我头大,所以整理本文以备日后查阅。 首先要知道上面提到的这些函数一部分是来自于torch.nn,而另一部分则来自于 ...

2019-02-19 23:12 0 38716
(原)PyTorch中使用指定的GPU

转载请注明出处: http://www.cnblogs.com/darkknightzh/p/6836568.html PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。 有如下两种方法来指定需要使用的GPU。 1. 类似tensorflow指定 ...

2017-05-11 00:20 6 83426
解读 pytorch对resnet的官方实现

地址:https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py 贴代码 import torch.nn as nn import torch.utils.model_zoo ...

2018-10-31 07:26 1 43672
anaconda下安装pytorch

anaconda下安装pytorch 1、 安装anaconda 官网一般显示的是最新版本:https://www.anaconda.com/products/individual 如果想要下载以往版本,可以在下面的网页:https://repo.anaconda.com/archive ...

2020-08-30 01:39 0 25938
PyTorch的nn.Linear()详解

1. nn.Linear() nn.Linear():用于设置网络中的全连接层,需要注意的是全连接层的输入与输出都是二维张量 一般形状为[batch_size, size],不同于卷 ...

2020-07-24 00:15 0 26583
pytorch版yolov3训练自己数据集

目录 1. 环境搭建 2. 数据集构建 1. xml文件生成需要Labelimg软件 2. VOC2007 数据集格式 3. 创建*.nam ...

2019-05-15 01:35 87 20945

 
粤ICP备18138465号  © 2018-2021 CODEPRJ.COM