/Image-Matching-v2

框架广告的图像匹配模型

Primary LanguagePythonMIT LicenseMIT

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.69% 19.82047499705621

角度 theta 的分布

image

False Positive

16 false positives:

小样图 上刊照(原图) 上刊照(校准)
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
False Negative

15 false negative:

小样图 上刊照(原图) 上刊照(校准)
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image