用于标记图像以进行分类的工具

数据挖掘 机器学习 图像分类 训练
2021-10-12 23:45:26

谁能推荐一个工具来快速标记数百张图像作为分类的输入?

我有大约 500 张细胞的显微镜图像。我想为训练集手动分配诸如'healthy''dead'、之类的类别,'sick'并将其保存到 csv 文件中。

基本上,与这个问题中描述的相同,除了我没有专有图像,所以也许这会开辟更多的可能性?

4个回答

我刚刚在 python 中编写了一个非常基本的助手,它要求所有图像都存储在一个 pyton 列表中allImages

import matplotlib.pyplot as plt
category=[]
plt.ion()

for i,image in enumerate(allImages):
    plt.imshow(image)
    plt.pause(0.05)
    category.append(raw_input('category: '))

pidgey - 具有主动学习功能的 jupyter notebook 小部件

我刚刚发现了这个开源工具,它看起来很棒: 在此处输入图像描述

它是 Jupyter Notebook 的交互式小部件,它的最佳之处 - 根据此提交 ,您可以添加自己的类似 sklearn 的分类器,并在注释时使用它来预测类!当您进行标记时,分类器会不断学习。


tkteach - 超快速图像分类 Python 工具

此外,还有一个名为tkteach的工具,它非常棒,因为您可以仅使用键盘快速注释图像。我对原始版本进行了一些改进。分叉在这里:https ://github.com/Serhiy-Shekhovtsov/tkteach

在此处输入图像描述

我已经创建了一个代码来满足您的需求,它在 GitHub 上以image-sorter2 的形式提供它不是“标记”图像,而是将图像放入一个新文件夹中,但是创建您正在谈论的 csv 是一个直接的扩展。与此处建议的其他脚本相比,image-sorter2 是 100% 免费的,您无需花时间绘制边界框 - 该脚本只需为您打开一个 GUI,您单击多个按钮之一并相应地单击每个图像被分类到所需的类文件夹中,例如“cats”、“dogs”、“trucks” aso

在此处输入图像描述

试试这个工具。它非常简单,完全符合您的要求 → 为给定文件夹中的图像分配标签。

https://github.com/robertbrada/PyQt-image-annotation-tool

在此处输入图像描述