图像分类任务实战

       本文基于AI研习社发布的“图像场景分类挑战赛”完成,主要是从一个小白如何去将一个图像分类的任务用代码跑起来的角度写的,也是自己的一个学习过程。

       图像场景分类挑战赛:https://god.yanxishe.com/97?from=god_home_list
       本文代码链接:基于预训练模型的图像分类代码模板

拿到比赛题目时,首先做以下几点观察:

  • 首先分析一下这个任务:就是一个将一些风景图片正确分类的简单任务。
  • 其次看一下官方给的数据集和标签:
    1. 数据都是世界各地的风景图片,共有6类,buildings、street、forest、sea、mountain、glacier。训练集有13627张图片,测试集有3407张图片,图片为RGB图片,格式为jpg。
    2. 标签文件为csv文件,内容为filename( ‘0.jpg’ )和label( ‘forest’ ),两列都是String类型的。
  • 然后看一下提交结果文件的格式:结果文件为csv文件,内容不需要Title,第一列为图片序号(0),Int类型,第二列为类别名称(‘str

你可能感兴趣的:(深度学习,图像分类,Pytorch)