zdyshine / Classification-merchant-signs

2018 Baidu merchant signboard classification and testing contest:2018百度商家招牌的分类与检测大赛

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Classification-merchant-signs

2018 Baidu merchant signboard classification and testing contest:2018百度商家招牌的分类与检测大赛

Overview

This is a summary of my participation in the classification and testing competition of 2018 Baidu merchant signs. Competition homepage: [百度](https://dianshi.baidu.com/competition/17/rule)。
The project runs on win10 + anaconada. You can also use other environments to run.
The program uses the InceptionV3 model for fine-tuning, and the final classification accuracy is 0.995.
I also tried to use resnet50, vgg16, Xception...they can also get a nice result.
As the project supervisor reason unable to attend the semi-finals

Network structure change

The final fully connected layer and category output layer of the network has changed according to the number of categories of actual classified items. You can also try to make different changes.

Installation requirements

python = 3.6.0
tensorflow >= 1.7.0
keras > = 2.1.3
argparse
matplotlib

Project file

data_pre.py : Divide the training set into a training set and a validation set.
data_arguement.py : Data enhancement,The default is to enhance 1 to 8 images.
finetune_model.py : Use this script to train.
finetune_model_test.py : Use this script to test.

Dataset file directory

Before enhancement
| datasets
| test |image1.jpg image2.jpg ...
| train
|image1.jpg image2.jpg ...
| test.txt
| train.txt

After enhancement
| datasets
| test
|image1.jpg image2.jpg ...
| train
| 1
|image1.jpg image2.jpg ...
| 2
|image1.jpg image2.jpg ...
... | valid
| 1
|image1.jpg image2.jpg ... | 2
|image1.jpg image2.jpg ...
...
| test.txt
| train.txt

Txt file format : image name + label.

Match Results

result

About

2018 Baidu merchant signboard classification and testing contest:2018百度商家招牌的分类与检测大赛

License:MIT License


Languages

Language:Python 100.0%