- 博客(3)
- 资源 (3)
- 收藏
- 关注
原创 windows下pytorch Dataloader中多进行重复执行代码的问题解决(重复创建tensorboard日志)
train = AudioFrameDataset('./datas/v2f_train.csv', transform=transform)trainloader = torch.utils.data.DataLoader(train, batch_size=B, num_workers=NUM_WORKERS, shuffle=True)举个例子,如上代码,dataloader里有一个...
2019-11-28 14:55:51 2695 4
原创 FGCROSSNET 代码分析
网络的主干部分使用ResNet-50网络的输入有四个,分别是图片,视频,音频,文本训练时网络的输出也有四个,分辨是四个输入所属的类别第一个loss为cross-entropy,分别对上述四个类别单独计算,然后求和第二个loss为center loss 乘以系数0.001,同时对上述四个进行计算,输入是完整的网络输出和四个输入类别的拼接(targets = torch.cat((targe...
2019-11-10 15:13:44 550
原创 使用自己的数据集训练Yolov3(Darknet53)
首先是先上yolo作者的网页,按照上面的提示把yolov3源码下载下载,github地址是https://github.com/pjreddie/darknet由于是用自己的数据集去训练网络,所以这里的预训练权重可以不用下载。然后就是去准备数据集的标签了。训练Darknet的标签主要是txt文件表示,每个图片对应一个txt文件,里面以下面这种格式保存这个图片上的每个bounding box....
2019-11-06 01:17:48 1586
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人