
利用PyTorch,One-Shot-Object-Detection 实现了具有共同注意力机制和共同激励机制的单阶段目标检测。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
提供了一项涵盖一站式目标检测的全面介绍,该介绍融合了共同注意机制和共同激励机制。谢婷一、罗以晨、陈焕宗和刘廷-在神经信息处理系统(NeurIPS)会议上发表了该成果,于2019年呈现。该项目专注于One-Shot Object Detection的纯PyTorch实现。 大量代码均来源于...我们目前正在进行以及计划进行的,是利用tensorboardX进行ImageNet预训练模型上传的支持。此外,我们还提供参考图片以及用于检查点的模型。为了训练PASCAL_VOC数据集,首先需要克隆代码:git clone https://github.com/timy90022/One-Shot-Object-Detection.git。 确保满足先决条件,包括Ubuntu 16.04操作系统、Python或3.6版本以及火炬1.02库。 数据准备方面,请参照可可中的详细说明来准备所需数据,并参考该存储库中提供的相关脚本。 在实验中,我们采用ResNet50作为预训练模型,并将其所有与COCO相关的层移除。
全部评论 (0)
还没有任何评论哟~


