Pytorch
关注
分享
扫一扫
文章平均质量分 56
LEILEI18A
这个作者很懒，什么都没留下…
展开
-
人脸姿态估计(二)之人脸检测-人脸关键点-人脸姿态任务合并
人脸姿态估计(二)之人脸检测-人脸关键点-人脸姿态任务合并0. 先放github链接https://github.com/gengyanlei/Pytorch_Retinaface# requirements:''' pytorch >= 1.1 python >= 3.5'''1. 目的将人脸检测、关键点(5个)检测、人...原创 2021-02-24 09:35:10 · 925 阅读 · 4 评论 -
语义分割数据增强python代码
语义分割数据增强python代码0. 先放github开源项目链接https://github.com/gengyanlei/segmentation_pytorch包含网络：deeplab_v3、deeplab_v3+、unet、pspnet、HF_FCN、U^2Net。。。不...原创 2021-02-20 17:06:01 · 2672 阅读 · 1 评论 -
tensorboard VS Weights & Biases
可视化工具tensorboard VS Weights & Biases1. tensorboard 安装使用1.1 安装# 安装 # pytorch1.1 pip install tensorboard==1.14 # pytorch1.6 pip install tensorboard==2.0 # pytorch1.7 pip install tensorboard==2.2# 使用 # 训练时，可在docker容.原创 2021-02-02 16:12:02 · 744 阅读 · 0 评论 -
安全帽反光衣检测识别数据集和yolov5模型
工作服(反光衣)检测数据集和yolov4-5检测模型目录0.摘要1.开源项目github链接2.数据集详细情况3.工作服(反光衣)数据集扩充方案0.摘要本文开源1个工作服(反光衣)检测数据集(含标注)和预训练模型，此项目已经上传github，欢迎star。马上更新1.开源项目github链接https://github.com/gengyanlei/...原创 2020-09-20 16:30:04 · 16666 阅读 · 35 评论 -
pytorch 模型并行 model parallel
pytorch model parallel 模型并行训练左侧：是网络太大，一张卡存不了，那么拆分，然后进行模型并行训练。右侧：多个显卡同时采用数据训练网络的副本。其实还有另一个：组卷积，每个组分到不同的GPU上，这样也可以做到并行。官方model parallel链接：https://pytorch.org/tuto...原创 2019-09-19 20:37:51 · 8771 阅读 · 6 评论 -
unet pytorch代码
github link: https://github.com/gengyanlei/U_Netpytorch ： 0.40 python3.5/3.6原创 2018-12-24 18:53:27 · 3353 阅读 · 0 评论 -
PSPNet Deeplab_v3+ pytorch复现
#pytorch：0.4 python：3.5 基于resnet50(PSPNet) or resnet50(Deeplab_v3+)更多操作github链接：https://github.com/gengyanlei''' PSPNet '''import torchfrom torch import nnimport torchvisionimport torch.n...原创 2018-06-15 10:59:47 · 6204 阅读 · 0 评论 -
pytorch cuda runtime error (59) 错误
pytorch训练过程中：出现cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70输出Variable(label).cuda()为空，在变成Variable之前 输出label最大最小值，一般情况绝对是超出了标签类别数。...原创 2018-06-12 20:29:55 · 5794 阅读 · 0 评论 -
Spp-Net Pytorch
基于Pytorch复现Spp-NetPytorch：0.4 ； Python：3.5。较tensorflow易实现。'''sppnet base on vgg16input size random ; but batch size need set to be 1 , and we don't use 'batch_size' .'''import torchimport torchvision...原创 2018-06-11 20:33:43 · 2261 阅读 · 0 评论 -
完整的pytorch教程（mnist 为例）
'''设置不同层的学习率、更新学习率（对应更新）、输出中间层特征、创建网络、保存网络、测试网络效果、初始化github link: https://github.com/gengyanlei/Pytorch-Tutorial-mnistA whole Pytorch tutorial : set different layer's lr , update lr (One to one ...原创 2018-05-28 21:47:12 · 5435 阅读 · 2 评论 -
Pytorch-HED fine-tune实现
Pytorch HED （VGG16-HED and Res34-HED ）python 3 ； pytorch 0.4基于 fine-tune 的 VGG16 或者 Resnet34 构建HED网络；并且，本人基于pytorch已经写好和训练好的网络基础上，构建属于自己简约风格的网络，例如HED网络。https://github.com/gengyanlei 链接 具体情况'''...原创 2018-05-16 17:01:23 · 2860 阅读 · 4 评论 -
pytorch 输出中间层特征
pytorch 输出中间层特征：tensorflow输出中间特征，2种方式：1. 保存全部模型(包括结构)时，需要之前先add_to_collection 或者 用slim模块下的end_points2. 只保存模型参数时，可以读取网络结构，然后按照对应的中间层输出即可。but：Pytorch 论坛给出的答案并不好用，无论是hooks，还是重建网络并去掉某些层，这些方法都不好用（在我看来）。我们可...原创 2018-05-21 11:20:57 · 17603 阅读 · 6 评论