文章目錄

• • • 模型
    • 加載數據集并訓練
    • 分類測試

模型

resnet18

18 = 17個卷積層 + 1個全連接層

流程

加載數據集并訓練

此demo是區分螞蟻和蜜蜂。

分類測試

【1】螞蟻

【2】蜜蜂

總結

以上是生活随笔為你收集整理的Pytorch demo（三）之蚂蚁和蜜蜂的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。