Yuqi's Lab7

README.md

@@ -1,4 +1,138 @@
 # lab07-machine-learning
+## Name
+Yuqi Zhang yuqiko 83234493
+
+## Classifier
+![](1.gif)
+
+// Copyright (c) 2019 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
+
+/* ===
+ml5 Example
+Webcam Image Classification using a pre-trained customized model and p5.js
+This example uses p5 preload function to create the classifier
+=== */
+
+// Classifier Variable
+
+let classifier;
+
+// Model URL
+
+let imageModelURL = 'https://teachablemachine.withgoogle.com/models/Am_wraDzH/';
+
+
+// Video
+
+let video;
+
+let flippedVideo;
+
+// To store the classification
+
+let cus_img;
+
+let label = "";
+
+// Load the model first
+
+function preload() {
+
+  classifier = ml5.imageClassifier(imageModelURL + 'model.json');
+
+}
+
+function setup() {
+
+  createCanvas(320, 260);
+
+  // Create the video
+
+  video = createCapture(VIDEO);
+
+  video.size(320, 240);
+
+  video.hide();
+
+  cus_img = createImg("https://media2.giphy.com/media/qPoCDRpGkRG5a/giphy.gif?cid=ecf05e47bxcrp7xscr0dw7b70lbtipfzx23sjy9reurrpd6f&rid=giphy.gif&ct=g");
+
+  cus_img.hide();
+
+  flippedVideo = ml5.flipImage(video)
+
+  // Start classifying
+
+  classifyVideo();
+}
+
+function draw() {
+
+  background(0);
+
+  // Draw the video
+
+  image(flippedVideo, 0, 0);
+
+  // Draw the label
+
+  fill(255);
+
+  textSize(16);
+
+  textAlign(CENTER);
+
+  if(label == "Class 1"){
+
+    image(cus_img, 0, 0, cus_img.width*0.75, cus_img.height*0.75);
+
+    text("Togepi", width / 2, height - 4);
+
+  }else{
+
+    text("Togekiss", width / 2, height - 4);
+
+  }
+
+}
+
+// Get a prediction for the current video frame
+
+function classifyVideo() {
+
+  flippedVideo = ml5.flipImage(video)
+
+  classifier.classify(flippedVideo, gotResult);
+
+}
+
+// When we get a result
+
+function gotResult(error, results) {
+
+  // If there is an error
+
+  if (error) {
+
+    console.error(error);
+
+    return;
+
+  }
+  // The results are in an array ordered by confidence.
+
+  // console.log(results[0]);
+
+  label = results[0].label;
+
+  // Classifiy again!
+
+  classifyVideo();
+}
+
+
 
 1. Create a brief demo with Teachable Machine
   - Go to https://teachablemachine.withgoogle.com/