我如何从一个图像集合，到一个完整的机器学习数据集，然后我可以使用它进行转移学习（resnet50等）？

Question

我是机器学习领域的新手，刚刚开始掌握这个领域的窍门。我在想我怎么能从大量的图像，变成一个机器学习数据集。我也不太清楚如何为我的数据存储标签？csv格式

Answer 1

这是对整个工作流的粗略描述。由于你的问题不太具体，我不能再详细说明了

1.为图像添加标签

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

我使用OpenLabeling以YOLO格式标记我的数据集。但是，这可能不是您想要的格式

2.找一个预先训练好的模特

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

3.安装电脑

你需要安装训练神经网络所需的一切。 我还建议您安装并使用CUDA（使用GPU而不是CPU来训练网络），因为它速度更快

4.训练你的模特

使用所选框架的转移学习功能，使用预先训练的模型和标记的图像来训练网络