pytorch
CCB_307
这个作者很懒,什么都没留下…
展开
-
pytorch自定义数据读取
pytorch的数据读取主要跟三个类有关: 1. Dataset 2. DataLoader 3. DataLoaderIter定义自己的Dataset类定义自己的Dataset类(继承torch.utils.data.Dataset)并实现两个成员方法 1. __getitem__() 2. __len__()class MyDataset(data.Dataset...原创 2018-09-02 00:50:43 · 825 阅读 · 0 评论 -
pytorch提取某一层网络特征图
pytorch提取某一层网络特征图由于pytorch的原模型很多用到了nn.Sequential,因此中间的特征图难提取。解决办法:自己写一个forward函数,然后索引需要的那一层。def resnet_cifar(net,input_data): x = net.conv1(input_data) x = net.bn1(x) x = F.relu(x) ...原创 2018-10-10 12:44:23 · 18196 阅读 · 5 评论