kaggle_whale_identification:Kaggle座头鲸识别挑战的代码

时间:2024-05-21 03:30:29
【文件属性】:

文件名称:kaggle_whale_identification:Kaggle座头鲸识别挑战的代码

文件大小:986.82MB

文件格式:ZIP

更新时间:2024-05-21 03:30:29

JupyterNotebook

kaggle_whale_identification 该存储库包含我针对Kaggle座头鲸识别挑战的代码。 给定鲸鱼吸食的图像,挑战在于预测鲸鱼的身份。 我当前应对这一挑战的方法是使用传递学习和Resnet18体系结构的预训练神经网络(NN)。 我的分析包含在以下文件中: data_exploration.ipynb :我的数据浏览表明,不同类别(aka类)的分布严重偏斜。 训练集包含近10000张图像和大约4000个不同的标签。 许多标签仅出现一次,而最常见的标签“ new_whale”出现810次,第二最常见的标签出现34次。 toy_model.ipynb :为了解决偏斜类的问题,我首先在由56个未偏斜的鲸鱼组成的子集中训练一个玩具模型。 该模型的目的是了解哪种特征与鲸鱼的识别有关。 full_model.ipynb :完整模型使用玩具模型的隐藏层,仅最后一层被更改。 希


网友评论