在Hsintao大佬代码的基础上进行了修改
与Hsintao大佬公布的结果做对比
Backbone | nme |
---|---|
MobileNetV2 | 4.96% |
MobileNetV3 | 4.40% |
Resnet50 | 4.03% |
- 数据集来源于飞桨社区
# 下载数据集到data/imgs下
cd data
python prepare.py
# data 文件夹结构
data/
imgs/
train_data/
imgs/
list.txt
test_data/
imgs/
list.txt
cd ..
python train.py