Code Monkey home page Code Monkey logo

yuhenghub / custom-vision-generator Goto Github PK

View Code? Open in Web Editor NEW

This project forked from nzigel/custom-vision-generator

0.0 0.0 0.0 1.44 MB

It's an end to end C# and Python sample from creating a project in custom vision till exporting confusion matrix in CSV format.

Home Page: https://medium.com/@nzigel/building-a-classification-model-for-android-ios-and-windows-ml-using-customvision-ai-5fa70d5b2303

C# 7.78% Jupyter Notebook 92.22%

custom-vision-generator's Introduction

Custom Vision Generator

A tool for generating a custom vision projects. It will automatically find and download images of topics you imput, train the model and export a confusion matrix in CSV format.

Pre-requisites

You'll need to install Visual Studio to build and run this code.

Steps to run the sample

  1. Sign in or create a new account on customvision.ai
  2. After login - click on cog icon on the right hand side and copy key values into the appropriate app settings

subscription keys

  1. Register for the free Bing Search service then copy the key value into the app settings

getting an API key

  1. Get a Computer Vision API key to perform the smart thumbnails of downloaded images

  2. Take a copy of SAMPLEApp.config rename it app.config and paste your keys into this file and set the size of your image set (~70% to train and 30% to test)

Running the application

The app will create a project in your custom vision dashboard with a random GUID. Please note that each time you run the app, a new project will be created. Next the app reads all the tags in the tags.csv file. The system searches Bing Images for each tag to prepare the source set of images for the classifier.

list of tags to download in tags.csv

By default it downloads ten images for each tag. This is set in the app.config as sizeOfImageSet. Ideally you should change this around 100 images to get better results. We perform image augmentation steps to increase the size of the training dataset. After downloading the images, the app will then upload the photos into the customvision.ai project and tag them accordingly.

Now the model has enough data to be trained. The app will train the model and set the default iteration to the newly trained model.

The model is ready for testing so the app will try to test the model the results of which are exported as a confusion matrix into result.csv. You can check the quality of your model by reviewing the confusion matrix.

As you are running the app you should see output similar to this animation. process running

More information on the Azure Custom Vision Service

custom-vision-generator's People

Contributors

azadehkhojandi avatar davidburela avatar jakkaj avatar nzigel avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.