
该项目提供基于集成深度CNN的多标签图像分类基准-源码。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
多标签图像分类任务借助集成深度卷积神经网络(CNN)进行基准测试,其代码说明已采用PyTorch 0.4版本进行验证。为了实现模型定制,可以通过移除相应的注释,从而对model1代码进行调整,以生成本文中提及的Model2(M2)和model3(M3)。 运行脚本时,请使用以下命令:`python resnet101_model1fc.py 1 512 16` (其中三个参数分别代表试验索引、补丁大小以及批处理大小)。值得注意的是,VOC2007数据集的评估指标与NUS-WIDE和MS-COCO数据集的评估指标存在差异,这是由于评估过程中忽略了包含“困难示例”的标注数据所致。 在模型训练过程中,我们采用所有可用的训练数据,并设定一个固定的停止标准来终止训练。 为了成功运行该代码,您可能需要从其官方网站获取三个数据集的图像资源。参考王谦、贾宁、Toby P. Breckon 发表的论文《使用集成深度CNN进行多标签图像分类的基线》,发表于2019年IEEE国际图像处理会议(台北)。
全部评论 (0)
还没有任何评论哟~


