车型识别

简介

使用pytorch做的车型分类模型，采用resnet网络，可以支持1777个类别

数据集使用的是hyperVID 下载地址为 链接: https://pan.baidu.com/s/1vvV2H5Jpewgba_VFsWvDcA 密码: vuo4

resnet_car.pth 为检测模型

predict.py 为预测程序

使用方法

pip install -r requirements.txt

数据预测

python predict.py

开始训练

下载数据集解压后执行 python dataset_split.py

分好后执行训练程序 python train.py

TODO

增加训练集