Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
.gitignore		.gitignore
MyDataset.py		MyDataset.py
README.md		README.md
ShuffleNetV2.py		ShuffleNetV2.py
best.pt		best.pt
latest.pt		latest.pt
test.py		test.py
train.py		train.py

Repository files navigation

小车装甲板识别（分类）

各个文件的功能

MyDataset.py：自定义数据集

ShuffleNetV2.py：在 ericsun99/Shufflenet-v2-Pytorch 源码的基础上改小了模型参数量

train.py：模型训练以及验证

test.py：模型测试

best.pt：在验证集上 Accuracy 最高的模型参数

latest.pt：最新训练完后的模型参数

About

No description, website, or topics provided.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%