📦 vertex-image-classifier

An npm package to make it easier to use vertex AI with Google autoML for image classification

✨ Features

• ✅ Integrated with Vertex AI / Google AIplatform
• ⚡ Ideal for classifying images from more complex datasets directly, letting vertex AI algorithms adjust hyperparameters and define the architecture

📥 Installation

npm install vertex-image-classifier
# or
yarn add vertex-image-classifier

Usage

const classifier = new VertexClassifier({
    projectId: "my-project-id",
    location: 'gc-location',
    datasetBucketName: 'my-bucket-name',
    vertexDatasetName: 'my-dataset-name',
    keyFilePath: 'path_to_account_service_credentials.json',
  });
await classifier.setupVertexClassifier();
await classifier.addToDataset("my-label-1", video_file1) //an File object of class video. The classifier will extract the respective frames and upload it to dataset bucket. 
await classifier.addToDataset("my-label-2", video_file2)

/* Trainig pipeline is divided in:
1- create a dataset in ai platform.
2- import data from the GCS storage bucket to dataset.
3- creates a training pipeline in ai platform with auto ML.
4- creates a endpoint for prediction
5- upload trained model to endpoint.
*/
await classifier.startTraining()