简体   繁体   English

使用 Tesnsorflow 训练图像分类 model

[英]Train an image classification model using Tesnsorflow

what image classification to use for training?用于训练的图像分类是什么?

we want to train a model with our own pictures for a school project.我们想用我们自己的图片为学校项目训练 model。 Can I use resnet50 or any other trainable model?我可以使用 resnet50 或任何其他可训练的 model 吗? I couldn't find Resnet50 trainable model and I found only pre-trained resnet50 model.我找不到 Resnet50 可训练的 model,我只找到了预训练的 resnet50 model。 We are using Tensorflow 2.我们正在使用 Tensorflow 2。

Thanks谢谢

You don't have to use the pre-trained Resnet.您不必使用预训练的 Resnet。 Load the keras version of resnet with weights=None and train the model yourself.使用 weights=None 加载 keras 版本的 resnet 并自己训练 model。

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM