python - 使用 Tesnsorflow 训练图像分类模型

Question

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

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

谢谢

标签： pythontensorflowtensorflow2.0training-dataresnet

Answer 1

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