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 的分布
False Positive
16 false positives:
小样图 | 上刊照(原图) | 上刊照(校准) |
---|---|---|
False Negative
15 false negative:
小样图 | 上刊照(原图) | 上刊照(校准) |
---|---|---|