区分柴犬和柯基的AI模型实践

上方模型若是未能加载出来，点击这里主动跳转open in new window

区分柴犬和柯基

一个小型的模型，用于区分柴犬和柯基。

大概细节

此模型基于18层的ResNet系列预训练模型，进行微调。
其中，首先冻结了除了分类头之外的17层，对分类头进行单独的训练，然后解冻所有层，进行整体的训练。

数据集

使用duckduckgo的图片搜索API，爬取了柴犬和柯基的图片进行训练，后续又手动排除了一些错误图片