PyTorch第一个图片识别模型
仿照PyTorch官网的衣服的例子自己写了个图片识别模型,算是自己的第一个PyTorch模型,记录下过程。
一路踩坑是免不了的。
自定义Dataset
官网给出的`CustomImageDataset`的代码大体可用,需要根据自己的图片存入路径和csv文件格式稍做修改。
Home of Little Jelly
仿照PyTorch官网的衣服的例子自己写了个图片识别模型,算是自己的第一个PyTorch模型,记录下过程。
一路踩坑是免不了的。
自定义Dataset
官网给出的`CustomImageDataset`的代码大体可用,需要根据自己的图片存入路径和csv文件格式稍做修改。
PyTorch开发运行部署。
本地
Docker下的
1. Python 3
2. JupyterLab
PyTorch的基础使用有以下几点。
1. Tensor
2. Dataset & DataLoader
3. Transform
PyTorch相关资源
1. [官网](https://pytorch.org)
平台不同,pytorch的安装方式也不同。