Google News
logo
ml5.js - Interview Questions
A convolutional neural network for image classification tasks
A convolutional neural network for image classification tasks : You can use convolutional neural networks in the ml5.neuralNetwork by setting the task:"imageClassification".
const IMAGE_WIDTH = 64;
const IMAGE_HEIGHT = 64;
const IMAGE_CHANNELS = 4;
const options = {
  task: 'imageClassification',
  inputs:[IMAGE_WIDTH, IMAGE_HEIGHT, IMAGE_CHANNELS],
  outputs: ['label']
}
const nn = ml5.neuralNetwork(options);
Advertisement