Skip to content

mritunjaysharma394/autoCustomVision

Use this GitHub action with your project
Add this Action to an existing workflow or create a new one
View on Marketplace

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

51 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Automating Image Classification with Microsoft Azure Custom Vision Training and Prediction

Build Status Open Source Love

This GitHub Action automates the entire process of image classification with Microsoft Azure Custom Vision Service.

How to use

Step 1:

To use the Custom Vision Service you will need to create Custom Vision Training and Prediction resources in Azure. To do so in the Azure portal, fill out the dialog window on the Create Custom Vision page to create both a Training and Prediction resource.

Step 2:

Get the Endpoint, Training Key, Prediction Key and Prediction Resource ID credentials andd save it as Secrets in your GitHub repository settings. You can find the items at the Custom Vision website . Sign in with the account associated with the Azure account you used to create your Custom Vision resources. On the home page (the page with the option to add a new project), select the gear icon in the upper right.

Image of above credentials

Save the above credentials with SECRET_NAME as: AZURE_ENDPOINT, AZURE_TRAINING_KEY, AZURE_PREDICTION_KEY, AZURE_PREDICTION_RESOURCE_ID respectively.

Step 3:

Upload the image dataset and Test File in the Repository. After pushing the changes, Copy the below workflow file and push it:

name: Auto Custom Vision Classifier
on:
  push:
    paths:
    - '**.yml'
jobs:
  build_model:
    runs-on: ubuntu-latest
    steps:
    - name: Automating Image Classification with Microsoft Azure Custom Vision Training and Prediction
      uses: mritunjaysharma394/[email protected]
      with:
        tags: "[Hemlock,Japanese Cherry]" # Rename it according to the folder name under images/ which will also be our name to the tags
        tagsVar: "[hemlock_,japanese_cherry_]" # Rename it according to the symmetry of file names under images/tag/ 
        trainSize: "10" # Train Size of Each Tag
        endpoint: ${{ secrets.AZURE_ENDPOINT }}
        trainingKey: ${{ secrets.AZURE_TRAINING_KEY }}
        predictionKey: ${{ secrets.AZURE_PREDICTION_KEY }}
        predictionResourceid: ${{ secrets.AZURE_PREDICTION_RESOURCE_ID }}