@xiaofeng1 在 中秋节,手把手训练一个识别月饼的CNN!🥮 中发帖

我之前写过《图片识别模型到底是怎么训练出来的?》。今天就是中秋节了,所以我写了一个识别月饼的模型手把手,训练出自己的模型。这篇不需要你提前懂CNN,跟着准备图片、运行代码,最后拿一张新图片考考模型就可以了! 

有提前训练好的模型best.pt,可以开箱即用


运行内存:运行内存至少8G
储存空间:至少剩余5G

第一步:安装Python和PyTorch
Python和PyTorch是训练模型必装的东西,所以我们要先安装。Python是运行代码需要的语言,PyTorch则帮我们计算CNN、Loss和反向传播。
如果你是Mac,并且安装了Homebrew(安装链接:Homebrew)在项目文件夹打开终端运行:
brew install python@3.12 && python3.12 -m venv .venv && .venv/bin/python -m pip instal...