Search code examples
pythontensorflowtensorflow2.0training-dataresnet

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. Can I use resnet50 or any other trainable model? I couldn't find Resnet50 trainable model and I found only pre-trained resnet50 model. We are using Tensorflow 2.

Thanks


Solution

  • You don't have to use the pre-trained Resnet. Load the keras version of resnet with weights=None and train the model yourself.