Ch04_Training_Tools

第4章 開源模型訓練工具

4.1 AI工作流程

4.1.1 問題定義

4.1.2 建立模型

模型動物園：

• Model Zoo
• Intel OpenVINO Toolkit (Github) - Open Model Zoo repository
  • Intel Pre-Trained Models
  • Public Pre-Trained Models
• ONNX Model Zoo
• Google TensorFlow Model Garden
• Kaggle - Models
• Papers with code - Methods
• Roboflow - Models
• 機器之心 - 算法

4.1.3 訓練調參

4.1.4 佈署推論

4.2 常見模型算法

4.2.1 影像分類

4.2.2 物件偵測

YOLO相關論文及連結：

• YOLOV1(CVPR 2016)
• YOLOV2(CVPR 2017)
• YOLOV3(Arxiv 2018)
• YOLOV4(Arxiv 2020)
• YOLOV5(Ultralytics 2021)
• YOLOV6(Meituan 2022)
• YOLOV7(Arxiv2022/CVPR2023)
• YOLOV8(Ultralytics 2022)

4.2.3 語義/實例分割

4.2.4 其它

4.3 開源訓練工具

4.3.1 Google Colab

4.3.2 Jupyter Notebook

欲直接執行Github中 *.ipynb 程式在Google Colab上的方式如下：

• 要先於瀏覽器中登錄Google帳號才使用Colab
• 一般於Github中 *.ipynb 名稱多半為 https://github.com/使用者名稱/專案名稱/blob/main/子目錄名稱/*.ipynb
• 將其改成下列型式即可直接啟動運行 https://colab.research.google.com/github/使用者名稱/專案名稱/blob/main/子目錄名稱/*.ipynb

4.3.3 PyTorch & OpenCV

OpenCV (Python) 參考範例：

• 【課程簡報】20210326_東南資科_創新應用實務02_OpenCV初體驗
• 【課程簡報】20210409_東南資科_創新應用實務03_OpenCV彩色影像處理
• 東南資科_創新應用實務_OpenCV範例源碼

延伸閱讀：

• 如何在Colab上顯示雲端硬碟(Google Drive)上的影像和視頻(含範例程式)
• OpenVINO與OpenCV搭配─幹電腦視覺的活一點都不累

4.3.4 ONNX

4.4 影像分類範例

完整範例程式：https://github.com/OmniXRI/NTUST_Colab_PyTorch_Classification

4.4.1 自定義模型訓練

Colab範例直接運行網址：
https://colab.research.google.com/github/OmniXRI/NTUST_Colab_PyTorch_Classification/blob/main/20201218_PyTorch_Classification_training.ipynb

4.4.2 自定義模型推論

Colab範例直接運行網址：
https://colab.research.google.com/github/OmniXRI/NTUST_Colab_PyTorch_Classification/blob/main/20201218_PyTorch_Classification_Inference.ipynb

4.4.3 模型儲存與還原

4.4.4 預訓練模型推論

Colab範例直接運行網址：
https://colab.research.google.com/github/OmniXRI/NTUST_Colab_PyTorch_Classification/blob/main/20201218_PyTorch_Classification_Pretrained_Inference.ipynb

參考文獻

• A Survey of the Recent Architectures of Deep Convolutional Neural Networks
• Benchmark Analysis of Representative Deep Neural Network Architectures
• Deep Learning for Generic Object Detection: A Survey
• A Survey on Deep Learning-based Architectures for Semantic Segmentation on 2D images
• 使用Google Colaboratory免費資源學AI，正是時候！
• 利用「深度學習」模型進行影像二元分類_簡報分享
• PyTorch Documentation
• OpenCV Online Documentation (4.4.0)
• 如何在樹莓派上用Python玩轉開源電腦視覺工具OpenCV