具有共同注意和共同激励的一站式目标检测
介绍
谢婷一,罗以晨,陈焕宗,刘廷- 神经信息处理系统(NeurIPS),2019年,
该项目是One-Shot Object Detection的纯pytorch实现。 大多数代码是从。
我们正在做和将要做的
支持tensorboardX
上载ImageNet预训练模型。
提供参考图片。
提供检查点模型。
训练PASCAL_VOC数据集
准备
首先,克隆代码
git clone https://github.com/timy90022/One-Shot-Object-Detection.git
1.先决条件
Ubuntu 16.04
Python或3.6
火炬1.0
2.数据准备
可可:也请按照中的说明准备数据。 请参阅此存储库中提供的脚本。
3.预训练模型
我们在实验中使用ResNet50作为预训练模型。 通过排除所有与COCO
1