Closed grewe closed 4 years ago
Do the following code lab to understand how you can take a pretrained model for a totally different dataset and retrain it for a new model with different classes.
OPTION1: if you are doing a classification model do this https://www.tensorflow.org/hub/tutorials/tf2_image_retraining
OPTION2: If you are doing a detection model do this https://colab.research.google.com/github/AlaaSenjab/-Tutorial-Tensorflow_Object_Detection_API_On_Custom_Dataset/blob/master/weapon_detection_BL.ipynb
Do the following code lab to understand how you can take a pretrained model for a totally different dataset and retrain it for a new model with different classes.
OPTION1: if you are doing a classification model do this https://www.tensorflow.org/hub/tutorials/tf2_image_retraining
OPTION2: If you are doing a detection model do this https://colab.research.google.com/github/AlaaSenjab/-Tutorial-Tensorflow_Object_Detection_API_On_Custom_Dataset/blob/master/weapon_detection_BL.ipynb