Banananator trainer

let bananasFed=0

let trashFed=0

​

let textY

​

let trash

let bananas

let trainButton

​

let classifier

let featureExtractor

​

function modelLoaded() {

  trash.drop(inputFileHandler('trash'))

  bananas.drop(inputFileHandler('banana'))

  trainButton.mouseClicked(trainButtonPressed)

  console.log('Model Loaded!');

}

function inputFileHandler(label){return (file)=>{

  if (file.type === 'image') {

    let inputImage = createImg(file.data, '');

    inputImage.hide();

    classifier.addImage(inputImage, label);

    if(label==='banana'){

      bananasFed++

    }else{

      trashFed++

    }

  } else {

    console.log('[Insert a rude comment about the input not being an image]');

  }

}}

function trainButtonPressed(){

  classifier.train((lossValue) => {

    if(lossValue){

  console.log('Loss is', lossValue);

    }else{featureExtractor.save()}

});

}

​

​

​

​

​

​

function setup() {

  featureExtractor = ml5.featureExtractor('MobileNet', modelLoaded);

  classifier=featureExtractor.classification()

  createCanvas(windowWidth, windowHeight);

  background('yellow');

  //boxes for dropping the training images

  let boxWidth=windowWidth/3

  let boxHeight=windowHeight/2

  let boxXmarigin=windowWidth/12

  let boxYmarigin=windowHeight/12 

  trash=createDiv()

  trash.position(boxXmarigin,boxYmarigin)

  trash.size(boxWidth,boxHeight)

  rect(boxXmarigin,boxYmarigin,boxWidth,boxHeight)

  bananas=createDiv()

  bananas.position(windowWidth/2+boxXmarigin,boxYmarigin)

  bananas.size(boxWidth,boxHeight)

  rect(windowWidth/2+boxXmarigin,boxYmarigin,boxWidth,boxHeight)

  //button for starting training

  let trainButtonWidth=200

  let trainButtonHeight=60

  let trainButtonYcenter=windowHeight*3/4

  trainButton=createDiv()

  trainButton.position(windowWidth/2-trainButtonWidth/2,trainButtonYcenter-trainButtonHeight/2)

  trainButton.size(trainButtonWidth,trainButtonHeight)

  rect(windowWidth/2-trainButtonWidth/2,trainButtonYcenter-trainButtonHeight/2,trainButtonWidth,trainButtonHeight)

  //text stuff

  textY=boxYmarigin+boxHeight/2

  textAlign(CENTER, CENTER);

  textSize(32);

  text('Train',windowWidth/2,trainButtonYcenter)

  text('Feed me!\n',windowWidth*3/4,textY)

  text('Trash for comparison\n',windowWidth/4,textY)

}

​

​

​

function draw() {

  strokeWeight(0)

  fill(255)

  rect(windowWidth*3/4-100,textY+20,200,40)

  rect(windowWidth/4-100,textY+20,200,40)

  fill(0)

  text(bananasFed,windowWidth*3/4,textY+40)

  text(trashFed,windowWidth/4,textY+40)

}