成功跑通reid_strong_baseline模型

配置环境:

1.模型下载:

git clone https://github.com/michuanhaohao/reid-strong-baseline.git

或访问michuanhaohao/reid-strong-baseline: Bag of Tricks and A Strong Baseline for Deep Person Re-identification (github.com)直接下载解压,(以上都不方便可以留下邮箱,私发)
2.下载ignite:

(访问GitHub - pytorch/ignite at v0.1.2下载),解压(不方便,则同上)
    cd到文件夹目录下,运行python setup.py install
3.数据集放置位置:放在data下
4. 下载并放置预训练权重:

将预训练权重(resnet50-19c8e357.pth,访问https://download.pytorch.org/models/resnet50-19c8e357.pth)放到model下(不方便下载则同上)
5. 修改配置文件: 修改configs下的4个文件的:MODEL.PRETRAIN_PATH为pth文件的绝对路径
6.训练指令(需根据您的实际环境修改参数,路径最好不要带中文,否则会有各种各样的错误):python train.py的绝对路径 --config_file='softmax_triplet.yml的绝对路径' MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('数据集名称')" DATASETS.ROOT_DIR "('data目录的绝对路径')" OUTPUT_DIR "('.\outputs\softmax_triplet.yml_outputs')" 

7.可能遇到的问题:模型太大内存分页不够,导致跑不动

修改建议:

(1)修改系统分配的页大小,可参考我的另一篇博客:OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading “C:\ProgramData\Anaconda3\lib\site-packages\to_Wsyoneself的博客-CSDN博客

(2)如果以上修改还是跑不动,或者卡住,可以修改配置文件(configs下4个yml文件)中的batch_size:需修改以下变量(参考将64改为16)
    cfg.SOLVER.IMS_PER_BATCH
    cfg.TEST.IMS_PER_BATCH
 运行时间较长,可以喝20杯咖啡等一等~

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: person_reid_baseline_pytorch是一个基于PyTorch框架的人员识别基线模型。它可以用于训练和测试人员识别模型,以识别不同人员之间的差异和相似之处。该模型提供了一些基本的功能,如数据加载、模型训练、模型测试等,可以帮助用户快速搭建和测试自己的人员识别模型。 ### 回答2: person_reid_baseline_pytorch是一个基于PyTorch框架的人体重识别模型。该模型采用卷积神经网络(Convolutional Neural Network,简称CNN)架构,采用多个卷积层和池化层进行特征提取,最后经过全连接层得到人物特征向量,实现了对人体进行识别。 该模型的训练数据集一般采用Market1501数据集,它包含超过1501个不同身份的超过32000张图片,是一个非常大且具有代表性的数据集,训练出的模型能够具有很好的泛化能力。 在训练过程中,除了使用标准的交叉熵损失作为损失函数外,还使用Triplet Loss作为损失函数的一种变形。该损失函数能够比较好地处理重识别中的最大间隔问题,并且能够实现对同一身份的多张图片之间的特征相似度比较。 对于测试集的预测部分,该模型采用的是余弦相似度进行特征向量相似度计算,相似度越高则说明两个图片中的人物越有可能是同一个人,可以实现良好的识别结果。 总的来说,person_reid_baseline_pytorch基于PyTorch框架实现的人体重识别模型,通过CNN架构进行特征提取,使用Triplet Loss进行训练,使用余弦相似度进行特征向量相似度计算,能够实现对人体进行高效准确的识别。 ### 回答3: person_reid_baseline_pytorch 是一个基于 PyTorch 框架的行人重识别算法基线模型,它实现了一个简单但功能齐全的行人重识别框架,可以训练和测试基于清晰照片的行人重识别任务。 该模型包含了基本卷积神经网络(CNN)的搭建和训练,其中包括几个关键部分:数据加载、模型定义、训练和测试。它使用数据增强技术和三元损失函数来训练模型,目的是使相同人的图像对与不同人的图像对之间的距离最大化。 person_reid_baseline_pytorch 采用 Market-1501 数据集进行测试,并在该数据集上取得了不错的结果。该数据集包含 1501 个行人,共有 12936 张图像,其中包括 6 种视角和多种行为。该算法读取图像数据并将之前处理后的行人图像输入神经网络,以便为每个图像计算行人的特征向量。然后,使用相关算法来比较图像对的距离,并确定它们是否属于同一人。该算法的性能指标包括准确度、精度和召回率等。 总之,person_reid_baseline_pytorch 提供了一个简单但完整的框架和基线模型,可以作为进一步解决行人重识别问题的基础。无论是学术界还是商业界,这种算法都有广泛的应用前景。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值