簡介:本課程以 Tensorflow.js 作為主要框架,覆蓋了實現垃圾分類 App 所需要的樣本準備、模型構建、模型訓練、模型預測、H5 應用編寫等全鏈路知識的講解。帶你在快速完成應用的同時,進一步加強對機器學習的認識!另,對機器學習以及Tensorflow.js感興趣的同學,可以關注一下老師的實戰《JavaScript玩轉機器學習 打造你人生中的第一個AI項目》https://coding.imooc.com/class/408.html 專為前端人打造的人工智能課程??!
第2章 用 Node.js 版 Tensorflow 訓練垃圾分類模型
本章主要講解如何使用 tensorflow.js 的 node 版本進行垃圾分類的模型訓練,包括:加載樣本、定義模型、訓練模型。其中,加載樣本就是將現實世界里的垃圾圖片和分類標簽轉成計算機認識的數據;定義模型則是建立一個神經網絡來判斷一個圖片是什么垃圾;訓練模型,是把樣本喂給模型,讓它學習。而在訓練完成后,則會生成一個模型文件,等待在 垃圾分類 App 里調用。
第3章 用 Tensorflow.js + React 編寫垃圾分類 H5 App
本章主要講解,如何將上一章訓練出來的模型加載到瀏覽器中進行垃圾圖片識別,包括:模型加載、模型預測以及H5 App 編寫。其中,模型加載就是使用 Tensorflow.js 提供的 API 將模型文件加載瀏覽器中;模型預測,則是將垃圾圖片轉為 tensor ,輸入到模型里,模型會輸出相應的分類,比如:有害垃圾或者廚余垃圾等。最后在H5 App 編寫階段,將會使用 Ant Design 組件庫去編寫 Html5 頁面,以便于在手機瀏覽器或者其他 App 里運行。