# pytorch-classification

**Repository Path**: liyuhai123/pytorch-classification

## Basic Information

- **Project Name**: pytorch-classification
- **Description**: 根据B站UP主的图像分类代码，自己修改整理和做实验的代码。
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2024-07-22
- **Last Updated**: 2025-06-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## 该文件夹存放使用pytorch实现的代码版本
**model.py**： 是模型文件  
**train.py**： 是调用模型训练的文件    
**predict.py**： 是调用模型进行预测的文件  
**class_indices.json**： 是训练数据集对应的标签文件   


------
若要使用该训练脚本需要下载对应的花分类数据集并将其划分为训练集和验证集。   
[点击这里](../data_set/README.md)会告诉你如何去下载数据集，以及提供了现成的划分数据集脚本  

出处来自这个up主：https://space.bilibili.com/18161609/?spm_id_from=333.999.0.0