使用 Tensorflow 训练图像分类模型

Question

用于训练的图像分类是什么？

我们想用我们自己的图片为一个学校项目训练一个模型。我可以使用 resnet50 或任何其他可训练的模型吗？我找不到 Resnet50 可训练模型，我只找到了预训练的 resnet50 模型。我们正在使用 TensorFlow 2。

谢谢

Answer 1

您不必使用预训练的 Resnet。使用 weights=None 加载 keras 版本的 resnet 并自己训练模型。