PyTorch
记录Python的学习之路
一个浪漫务实的菜鸟
AI、计算机视觉
展开
-
retianet + pytorch-reid实现了在摄像头中行人重识别的任务
说明代码地址:https://github.com/zhanghlgithub/pytorch-reid.git本代码完成了行人检测和行人重识别的工作,行人检测使用Retianet作为主干网络,损失函数使用Focal loss,参考论文行人重识别,参考pytorch-reid库,参考代码。本代码修改了库的部分源码,实现了在摄像头中实时的响应行人重识别的任务。流程图如下:目录说明- d...原创 2020-01-02 09:51:31 · 2696 阅读 · 2 评论 -
pytorch-cifar10 对不同模型的测试
说明代码链接:https://github.com/zhanghlgithub/pytorch-backbone-cifar10为了方便快速学习和测试不同网络结构的性能表现,本模块封装好了接口。公开数据集基于cifar10数据测试。方便做实验和学习最新的方法。使用方法在使用时,只需把自己学到的最新网络结构添加到backbone模块即可,然后修改训练脚本train.sh中的backbone参...原创 2019-12-16 16:45:20 · 398 阅读 · 0 评论 -
Pytorch 1:torch.nn.Conv2d()函数详解
函数原型torch.nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0,dilation=1,groups=1,bias=True,padding_mode='zeros')参数说明: in_channels(int) – 输入图像中的通道数 out_channels(in...原创 2019-12-10 22:38:02 · 1148 阅读 · 0 评论