pytorch实战 猫狗大战Kaggle 迁移学习ResNet50模型微调

2021-04-11 06:25

阅读:635

  • 这是kaggle上一个非常经典的二分类图像数据集,训练集包括25000张猫和狗的图片及其标签,测试集则是12500张未标签图片,数据下载地址https://www.kaggle.com/c/dogs-vs-cats/data。不过这个网址比较远古,无法提交自己训练的答案,可以到新的(虽然也不新了)比赛链接提交https://www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition/overview

  • 将训练数据按类别分开整理成如下结构

    |-- train 
        |-- cat
            |-- 1.jpg
            |-- 2.jpg
            |-- ...
        |-- dog
            |-- 1.jpg
            |-- 2.jpg
            |-- ...


评论


亲,登录后才可以留言!