首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >使用coco训练的图像分类模型

使用coco训练的图像分类模型
EN

Stack Overflow用户
提问于 2020-12-08 05:41:49
回答 1查看 187关注 0票数 0

我需要使用使用名为coco的数据集训练的图像分类。我在网上搜索了一下,但只找到了物体探测器。

有人知道tensorflow 2的预训练模型图像分类吗?

EN

回答 1

Stack Overflow用户

发布于 2020-12-08 19:13:29

目标检测不同于图像分类。

目标检测算法是图像分类和目标定位的结合。它接受一幅图像作为输入,并生成一个或多个边界框,每个边界框上都附加了类标签。这些算法有足够的能力处理多类分类和定位,以及处理多次出现的对象。ref

但是,您可以使用coco数据集剥离每个本地化框并创建一个新的数据集,该数据集现在可以为您的图像分类器提供数据。为了实现这一点,您必须对coco数据集进行处理。

例如。如果你的数据集注释看起来像这样:(包含4个本地化的对象,如下所示):

kite image_1.jpg

  • 133,123,23,13,45,kite,kite,43,213,77,image_1.jpg
  1. 133,13,413,73,鸟,image_2.jpg
  2. 12,233,440,34,,树,image_1.jpg

您可以编写脚本将其转换为以下内容:

  1. kite_123231345image_1.jpg
  2. bird_1334521377image_1.jpg
  3. bird_1331341373image_2.jpg
  4. tree_1223344034image_1.jpg

并使用剪切框创建这些图像。

并使用此注释来训练您的分类器。我个人以前在cocodataset上做过这件事,并且给出了不错的精确度。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65189843

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档