狗狗识别——基于ViT的犬类品种分类开源项目

我要开发同款
ManiaAmaeOvo2025年08月30日
56阅读

技术信息

语言技术
PythonHTML5CSSTorch
开源地址
https://github.com/ManiaAmaeOvo/vit_dogs_classifier
授权协议
MIT许可

功能介绍

本项目使用Google提供的vit-base-patch16-224-in21k模型进行微调,完成了犬类品种分类任务。

训练时长: 5个epochs。
深度学习框架: PyTorch。
前后端实现: Flask 和 HTML。
通过简单的操作,即可在本地端口5050访问前端WebUI,并拖拽图片实现犬类品种的识别。

示例图片

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论