README.md
classification
物体识别分类
项目介绍
该项目对物体进行识别分类。
项目配置
- 作者开发环境:
- Python 3.7
- PyTorch >= 1.5.1
数据集
- 采用"Stanford Dogs Dataset"数据集官方地址:http://vision.stanford.edu/aditya86/ImageNetDogs/
- 本文将该数据集的标注文件更改为xml格式,数据集下载地址(百度网盘 Password: ks87 )
- 通过运行read_datasests.py,可以对数据的标注信息进行解析可视化。
预训练模型
项目使用方法
模型训练
- 根目录下运行命令: python train.py (注意脚本内相关参数配置 )
模型推理
- 根目录下运行命令: python inference.py (注意脚本内相关参数配置 )
注意事项
- 因为数据集的整张图会有大量背景、多只狗等干扰因素。本文采用的图像输入是"Stanford Dogs Dataset"数据集的目标边界框范围图像,这样也觉定了在模型推断预测环境下也是以目标物的边界框为范围的图像作为输入。