

Deploy YOLOv5 Classification Models to GCP Compute Engine

Deploy your models with Roboflow

Hosted or on-device deployment

SDKs optimized for maximum performance

Extensive documentation

Get started for free

Learn More

How to Deploy YOLOv5 Classification Models to GCP Compute Engine

In this guide, we are going to show how to deploy a

YOLOv5 Classification

model to

GCP Compute Engine

using Roboflow Inference. Inference is a high-performance inference server with which you can run a range of vision models, from YOLOv8 to CLIP to CogVLM.

To deploy a

YOLOv5 Classification

model to

GCP Compute Engine

, we will:

1. Set up our computing environment
2. Download the Roboflow Inference Server
3. Try out our model on an example image

Let's get started!

In this guide, we are going to show how to deploy a

YOLOv5 Classification

model to

GCP Compute Engine

using the Roboflow Inference Server. This SDK works with

YOLOv5 Classification

models trained on both Roboflow and in custom training processes outside of Roboflow.

To deploy a

YOLOv5 Classification

model to

GCP Compute Engine

, we will:

1. Train a model on (or upload a model to) Roboflow
2. Download the Roboflow Inference Server
3. Install the Python SDK to run inference on images
4. Try out the model on an example image

Let's get started!

Train a Model on or Upload a Model to Roboflow

If you want to upload your own model weights, first create a Roboflow account and create a new project. When you have created a new project, upload your project data, then generate a new dataset version. With that version ready, you can upload your model weights to Roboflow.

Download the Roboflow Python SDK:

pip install roboflow

Then, use the following script to upload your model weights:

from roboflow import Roboflow

home = "/path/to/project/folder"

rf = Roboflow(api_key=os.environ["ROBOFLOW_API_KEY"])
project = rf.workspace().project("PROJECT_ID")

project.version(PROJECT_VERSION).deploy(model_type="yolov5", model_path=f"/{home}/yolov5/runs/train/")

You will need your project name, version, API key, and model weights. The following documentation shows how to retrieve your API key and project information:

- Retrieve your Roboflow project name and version
- Retrieve your API key

Change the path in the script above to the path where your model weights are stored.

When you have configured the script above, run the code to upload your weights to Roboflow.

Now you are ready to start deploying your model.

Set up a GCP Compute Engine Virtual Machine

Open GCP Compute Engine and click the “Create Instance” button to create a virtual machine.

Next, you need to configure your instance. The requirements for configuration depend on your use case. If you are deploying a server for production, you may opt for a more powerful machine configuration. If you are testing a model and plan to deploy on another machine in the future, you may instead opt to deploy a less powerful machine.

You can deploy Roboflow Inference on CPU and GPU devices. We recommend deploying on GPU for the highest performance. But, GPU devices are most costly to run and there is additional setup associated with using GPUs. For this guide, we will focus on CPUs.

A cost panel will appear on the right of the screen that estimates the cost of the machine you are deploying.

Fill out the required fields to configure your virtual machine. Then, click the “Create” button to create a virtual machine. It will take a few moments before your machine is ready. You can view the status from the Compute Engine Instances page.

When your virtual machine has been deployed, click on the machine name in the list of virtual machines on the Compute Engine Instances page.

To sign in using SSH in a terminal, click the arrow next to the SSH button and click “View gcloud command” If you have not already installed gcloud, follow the gcloud installation and configuration instructions to get started.

‍

Download the Roboflow Inference Server

The Roboflow Inference Server allows you to deploy computer vision models to a range of devices, including

GCP Compute Engine

.

The Inference Server relies on Docker to run. If you don't already have Docker installed on the device(s) on which you want to run inference, install it by following the official Docker installation instructions.

Once you have Docker installed, run the following command to download the Roboflow Inference Server on your

GCP Compute Engine

.

‍

Now you have the Roboflow Inference Server running, you can use your model on

GCP Compute Engine

Install the Roboflow Python SDK

The Roboflow Inference Server provides a HTTP API with a range of methods you can use to query your model and various popular models (i.e. SAM, CLIP). You can read more about all of the API methods available on the Roboflow Inference server in the Inference Server documentation.

The Roboflow Python SDK provides abstract convenience methods for interacting with the HTTP API. In this guide, we will use the Python SDK to run inference on a model. You can also query the HTTP API itself.

To install the Python SDK, run the following command:

pip install roboflow

Run Inference on an Image

With the Python SDK installed, you can run inference on your model in a few lines of Python code.

The following code will run inference on the model hosted on Roboflow and return a JSON object with predictions:

from roboflow import Roboflow
rf = Roboflow(api_key="API_KEY")
project = rf.workspace().project("PROJECT_NAME")
model = project.version(MODEL_VERSION, local="http://localhost:9001").model

# infer on a local image
print(model.predict("your_image.jpg", confidence=40, overlap=30).json())

# visualize your prediction
# model.predict("your_image.jpg", confidence=40, overlap=30).save("prediction.jpg")

# infer on an image hosted elsewhere
# print(model.predict("URL_OF_YOUR_IMAGE", hosted=True, confidence=40, overlap=30).json())

Substitute the model name, version, and API key with the values associated with your Roboflow account and project, then run the script.

‍

Enterprise-grade security and compliance

We take security seriously and have implemented comprehensive measures to keep your sensitive data safe



Compliant with SOC2 Type 1 requirements



All data is encrypted in transit and at rest, with SSL transport receiving a grade A+ rating from Qualys



Strict row-level permissions to ensure users cannot access sensitive data outside of their organizations



Roboflow is hosted on the Google Cloud Platform and Amazon Web Services, best-in-class infrastructure as a service providers



Authentication, database and file storage mechanisms are ISO 27001, ISO 27017, ISO 27018, SOC 1, SOC 2 and SOC 3 compliant



PCI compliant with Self-Assessment Questionnaire A and Attestation of Compliance



All card numbers and bank accounts never touch our servers and are stored by Stripe, a PCI Service Provider Level 1, the highest available security certification in the payments industry



Access to production data is heavily restricted within Roboflow and only accessible via SSO login



All Roboflow employees sign nondisclosure agreements restricting them from sharing information learned while handling customer data



Learn more about Roboflow for enterprise

Learn how to deploy models to other devices

Below, you can find our guides on how to deploy

YOLOv5 Classification

models to other devices.

Deploy YOLOv5 Classification Models to Azure Virtual Machines

Deploy YOLOv5 Classification Models to AWS EC2

Deploy YOLOv5 Classification Models to GCP Compute Engine

Deploy YOLOv5 Classification Models to Raspberry Pi

Documentation

The following resources are useful reference material for working with your model using Roboflow and the Roboflow Inference Server.