/AnimalClassification

卷积神经网络ResNet进行动物10分类

Primary LanguagePython

ResNet系列动物10分类网络

1.Requirement

torchvision >= 0.13.1
torch>= 1.12.1+cu116
opencv-contrib-python >= 4.6.0.66
numpy >=1.23.3
scikit-learn >= 1.1.2

2.本项目采用Pytorch编写,尽量采用简洁的模块设计没有多余的代码,并写有大量的注释 方便入门ResNet的使用及卷积神经网络的的训练

数据集采用10类动物进行训练 数据集下载地址:链接:https://pan.baidu.com/s/1brkneUrwEmgmpIFFajmNvQ 提取码:6666

根据训练的调整,ACC可以轻松超过0.95以上