PyTorch第一个图片识别模型

仿照PyTorch官网的衣服的例子自己写了个图片识别模型,算是自己的第一个PyTorch模型,记录下过程。

一路踩坑是免不了的。

自定义Dataset

官网给出的`CustomImageDataset`的代码大体可用,需要根据自己的图片存入路径和csv文件格式稍做修改。

豫ICP备19001387号-1