# PytorchClas

**Repository Path**: culture-studio/PytorchClas

## Basic Information

- **Project Name**: PytorchClas
- **Description**: 计算机视觉技术--图像分类算法汇总--使用Pytorch算法框架

- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 0
- **Created**: 2023-05-20
- **Last Updated**: 2024-06-26

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

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

------
结构目录

```
	|-- dataset        # 存放数据集的文件夹
	|---- train        # 训练集
	|---- val          # 验证集
    |-- other          # 附件文件夹
    |-- Pretrain_Model  # 预训练模型存放文件夹
    |-- Test1_official_demo   # 官方提供的测试代码
    |-- Test2_alexnet         # alexnet网络
    |-- Test3_vggnet          # vgg16net网络
    |-- Test4_googlenet       # googlenet网络
    |-- Test5_resnet          # resnet网络
    |-- Test6_mobilenet       # mobilenet网络
    |-- class_indices.json    # 分类类别的json文件
    |-- split_data.py         # 随机划分数据集比例的脚本
    |-- show_img_Total_loss_acc.py    # 展示不同网络的loss和Acc曲线
	
```

各网络对比结果

![Val_Accuracy](output\Val_Accuracy.png)

![Loss_res](output\Loss_res.png)