Dion van Beelen dvbeelen

## retrain-mobilenet-for-the-web.md

      
        
          
            
              
              1 file
            
          
          
            
              
              5 forks
            
          
          
            
              
              6 comments
            
          
          
            
              
              17 stars
            
          
        
        
          
              
          
          
            
                woudsma
                / retrain-mobilenet-for-the-web.md
            
            
              Last active
              November 1, 2022 05:09
            
              
                Retrain a MobileNet model for the web with TensorFlow for Poets and TensorFlow.js
              
          
        
      
        
  
      
    Retrain a MobileNet model and use it in the browser with TensorFlow.js

DRAFT

Combining TensorFlow for Poets and TensorFlow.js.

Retrain a MobileNet V1 or V2 model on your own dataset using the CPU only.

I'm using a MacBook Pro without Nvidia GPU.
MobileNets can be used for image classification. This guide shows the steps I took to retrain a MobileNet on a custom dataset, and how to convert and use the retrained model in the browser using TensorFlow.js. The total time to set up, retrain the model and use it in the browser can take less than 30 minutes (depending on the size of your dataset).
Example app - HTML/JS and a retrained MobileNet V1/V2 model.