使用 Tensorflow 训练图像分类模型

Question

使用什么图像分类进行训练？

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

谢谢

Answer 1

您不必使用 pre-trained Resnet。用weights=None加载keras版本的resnet，自己训练模型。