使用 JavaScript 进行图像分类

图像分类是机器学习最常见的应用之一。图像分类是一种计算机视觉技术，我们根据图像中的视觉内容对图像进行分类。我们可以训练一个图像分类器的示例，该分类器可以预测给定图像是否是狗。在本文中，我们将使用 ml5 js，它是一个用于 Web 的机器学习库。

迁移学习：在机器学习中，它是一种非常流行的技术。在这项技术中，我们使用为一项任务训练的机器学习模型，用于另一项类似任务。在这里，我们将使用 Mobile Net 一个预训练的机器学习模型来完成图像分类任务。

• HTML 代码：在这里，我们创建了一个 html 文件并使用ml5 js cdn在我们的文件中使用它。我们创建了一个输入字段来上传图像文件。 onchange事件侦听器设置为loadFile()函数，我们将在单独的 javascript 文件中定义该函数。此外，还添加了一个空白图像字段以显示上传的图像。
  索引.html：

      
          
          
          Image Classifier using ML5 js
          
          
      
      
          
              GeeksforGeeks
              
                  Image Classification using Javascript
              
              
              
              
              
              Predict
          
      

• JS 代码：如果您更喜欢外部 js 文件，请在script.js文件中创建一个 loadFile函数。首先，从 HTML 文件的输入字段中获取图像，并将图像设置在空白图像字段中。现在我们将启动我们的机器学习模型。然后调用 ml5.imageClassifier 方法，传递一个参数“MobileNet” ，以便为迁移学习加载 MobileNet，第二个参数是一个回调函数，记录“模型加载！”加载模型时在控制台中。现在在 js 部分创建一个 predict函数并调用classifier.predict 方法。第一个参数是要分类的图像，第二个参数是回调函数。现在我们使用alert方法来展示模型的预测结果。单击预测按钮时将调用此函数。

        var loadFile = function (event) {
        var image = document.getElementById("image");
        image.src = URL.createObjectURL(event.target.files[0]);
         };
        const classifier = ml5.imageClassifier
         ("MobileNet", modelLoaded);
    
         // When the model is loaded
         function modelLoaded() {
         console.log("Model Loaded!");
         }
                
         function predict() {
         classifier.predict(document.getElementById("image"), 
                  function (err, results) {
                      alert(results[0].label);
                  });
              }

最终解决方案：在本节中，我们只是将上述两个代码部分合二为一，以实现

    
        
        
        Image Classifier using ML5 js
        
        
     
        var loadFile = function (event) {
        var image = document.getElementById("image");
        image.src = URL.createObjectURL(event.target.files[0]);
       };
       const classifier = ml5.imageClassifier
       ("MobileNet", modelLoaded);
  
       // When the model is loaded
       function modelLoaded() {
       console.log("Model Loaded!");
       }
              
       function predict() {
       classifier.predict(document.getElementById("image"), 
              function (err, results) {
                    alert(results[0].label);
                });
            }
  
        
    
    
        
            GeeksforGeeks
            
                Image Classification using Javascript
            
            
            
            
            
            Predict
        
    

输出：