Image classification is a task where the model assigns a label to an image based on its content, such as “stop sign” or " speed limit " 1 . Object detection is a task where the model locates and identifies multiple objects in an image, and draws bounding boxes around them 2 . Since your dataset consists of images that were cropped to show one out of ten different traffic signs, you are dealing with an image classification problem, not an object detection problem. Therefore, you need to train a model for image classification, not object detection.
Vertex AI AutoML is a service that allows you to train and deploy high-qual ity ML models with minimal effort and machine learning expertise 3 . You can use Vertex AI AutoML to train a model for image classification by uploading your images and labels to a Vertex AI dataset, and then launching an Au toML training job 4 . However, Vertex AI AutoML does not allow you to tune the model during each training run, as it automatically selects the best m odel architecture and hyperparameters for your data 4 .
Vertex AI custom tr aining is a service that allows you to train and deploy your own custom ML models using your own code and frameworks 5 . You can use Vertex AI custom training to train a model for image classification by writing your own model training code, such as using TensorFlow or PyTorch, and then creating and running a custom training job. Vertex AI custom training allows you to tune the model during each training run, as you can specify the model architecture and hyperparameters in your code, and use Vertex AI Hyperparameter Tuning to optimize them .
Therefore, the best option for your scenario is to develop the model training code for image classification and train a model by using Vertex AI custom training.
[:, Image classification | TensorFlow Core, Object detection | TensorFlow Core, Introduction to Vertex AI AutoML | Google Cloud, AutoML Vision | Google Cloud, Introduction to Vertex AI custom training | Google Cloud, [Custom training with TensorFlow | Vertex AI | Google Cloud], [Hyperparameter tuning overview | Vertex AI | Google Cloud], ]
Submit