- 博客(5)
- 收藏
- 关注
RSS订阅原创 Pytorch自定义网络结构+读取自己数据+自定义Loss 全过程代码示例
1、限定使用GPU的序号import osos.environ['CUDA_VISIBLE_DEVICES'] = '3'os.system('echo $CUDA_VISIBLE_DEVICES')2、导入相关头文件import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.utils.dat...
2018-05-25 17:37:49
14939
7
原创 Pytorch如何自定义Loss
将Loss视作单独的层，在forward函数里写明loss的计算方式，无需定义backwardclass MyLoss(nn.Module): def __init__(self): super(MyLoss, self).__init__() print '1' def forward(self, pred, truth): ...
2018-05-25 17:22:01
10657
2
原创 如何用Pytorch提取视频单帧网络特征+SVM分类 - 非End-to-End
1、提取网络特征import osos.environ['CUDA_VISIBLE_DEVICES'] = '3'os.system('echo $CUDA_VISIBLE_DEVICES')import torchimport torch.nn as nnfrom torchvision import transforms, modelsfrom torch.autograd im...
2018-05-24 20:54:07
4963
2
原创 如何用Pytorch版本的FasterRCNN训练自己的数据
1、从这里下载FasterRCNN的Pytorch版本代码。按照提示完成相关编译。2、因为默认的数据为VOC2007，最直接的方法我们可以按照这里将自己的数据制作成VOC2007的格式，并建立软连接。3、修改https://github.com/jwyang/faster-rcnn.pytorch/blob/master/lib/datasets/pascal_voc.py数据的类别，即将self...
2018-05-15 20:12:21
6754
4
原创 制作自己的VOC2007格式数据
本质就是建立三个文件夹，Annotation用于存放xml标记文件，JPEGImages用于存放image数据，ImageSets文件夹下的Main文件夹存放train.txt,val.txt,trainval.txt,test.txt四个list。代码如下：import osdef dir2txt(dir, txt): files = os.listdir(dir) fw =...
2018-05-10 17:00:10
1998
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人