Train an image classification model using Tesnsorflow

Issue

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.

Answered By – Jithin Thenasseril

Answer Checked By – Marie Seifert (AngularFixing Admin)

Leave a Reply

Your email address will not be published.