我如何从图像集合到完整的机器学习数据集,然后我可以将其用于迁移学习(resnet50 等)?

How do I go from a collection of images, to a full machine learning dataset, that I can then use for transfer learning (resnet50 etc.)?

我是机器学习领域的新手,刚刚开始掌握它。我想知道如何从大量图像到机器学习数据集。我也不太确定如何为我的数据存储标签? csv 格式?

这是对完整工作流程的粗略描述。我不能提供更多细节,因为你的问题不是很具体。

1。标记您的图片

有很多工具可以标记一组图像。您需要找到一个适合您想要用来训练神经网络的深度学习框架的框架。

我已经使用 OpenLabeling 以 YOLO 格式标记我的数据集。但是,这可能不是您要查找的格式。

2。查找预训练模型

如果你想做迁移学习,你需要一个已经在另一个数据集上训练过的模型(神经网络)。

3。设置您的电脑

您需要安装训练神经网络所需的一切。 我还建议您安装和使用 CUDA(使用您的 GPU 来训练网络而不是您的 CPU),因为它要快得多。

4。训练您的模型

使用您选择的框架的迁移学习功能,使用预先训练的模型和您标记的图像来训练您的网络。