foamliu / Image-Matching

框架图像识别

Image-Matching

数据集

共计 9935 个方案，373471个照片。

依赖

• Python 3.6.8
• PyTorch 1.3.0

如何使用

数据准备

数据解压缩并预处理:

$ python extract.py
$ python pre_process.py

训练

$ python train.py

可视化训练过程：

$ tensorboard --logdir=runs

效果评估

测试数据集

开始评估

$ python test.py

准确度

Accuracy	Threshold
99.78%	21.07971786746929

角度 theta 的分布

False Positive

18 false positives:

小样图	上刊照

False Negative

4 false negative:

小样图	上刊照