Github | 基于RetinaNet的半自动图像标记工具

2019 年 10 月 28 日 机器学习算法与Python学习
本库的作者 Viraj Mavani 提供了一个新的图像注释工具 该工具包含一个名为 RetinaNet 的现有最先进物体检测模型,来显示并注释常用的 80 个对象类的建议,以此来减少开发者的负担。
Github 链接:
https://github.com/virajmavani/semi-auto-image-annotation-tool

安装

  1. Clone 本库

  2. 在库中,执行 pip install . --user。请注意,由于安装 Tensorflow 的方法不同,该软件包并未定义对张量流的依赖性,不过它会尝试安装(至少在 Arch Linux 导致错误的安装结果的时候)。

  3. 下载预训练的权重并将其保存在/快照中。

依赖

  1. Tensorflow >= 1.7.0

  2. OpenCV >= 3.4.0

  3. Keras >= 2.1.3

  4. For, Python >= 3.5

用法

   
   
     
python main.py
  1. 从下拉菜单中选择需要建议的COCO对象类,然后添加它们。

  2. 手动注释时,从列表中选择对象类,并保持选中状态,然后选择BBox。

  3. 最终的注释可以annotations.csv在./annotations/ 中的文件中找到

推荐阅读
7 个 Python 特殊技巧,有效提升数分效率!
年薪70W架构师:全套英语学习资源泄露,手慢则无!(禁止外传)
编程大神进阶之路:Python技巧小贴士
AI智能代码补全:减少一半操作,实时补全,免费下载!
登录查看更多
1

相关内容

RetinaNet是2018年Facebook AI团队在目标检测领域新的贡献。它的重要作者名单中Ross Girshick与Kaiming He赫然在列。来自Microsoft的Sun Jian团队与现在Facebook的Ross/Kaiming团队在当前视觉目标分类、检测领域有着北乔峰、南慕容一般的独特地位。这两个实验室的文章多是行业里前进方向的提示牌。 RetinaNet只是原来FPN网络与FCN网络的组合应用,因此在目标网络检测框架上它并无特别亮眼创新。文章中最大的创新来自于Focal loss的提出及在单阶段目标检测网络RetinaNet(实质为Resnet + FPN + FCN)的成功应用。Focal loss是一种改进了的交叉熵(cross-entropy, CE)loss,它通过在原有的CE loss上乘了个使易检测目标对模型训练贡献削弱的指数式,从而使得Focal loss成功地解决了在目标检测时,正负样本区域极不平衡而目标检测loss易被大批量负样本所左右的问题。此问题是单阶段目标检测框架(如SSD/Yolo系列)与双阶段目标检测框架(如Faster-RCNN/R-FCN等)accuracy gap的最大原因。在Focal loss提出之前,已有的目标检测网络都是通过像Boot strapping/Hard example mining等方法来解决此问题的。作者通过后续实验成功表明Focal loss可在单阶段目标检测网络中成功使用,并最终能以更快的速率实现与双阶段目标检测网络近似或更优的效果。
深度神经网络实时物联网图像处理,241页pdf
专知会员服务
76+阅读 · 2020年3月15日
专知会员服务
109+阅读 · 2020年3月12日
专知会员服务
41+阅读 · 2020年2月20日
【Google AI】开源NoisyStudent:自监督图像分类
专知会员服务
54+阅读 · 2020年2月18日
【GitHub实战】Pytorch实现的小样本逼真的视频到视频转换
专知会员服务
35+阅读 · 2019年12月15日
开源OCR文本检测器,基于TextBoxes++和RetinaNet
专知
11+阅读 · 2019年11月15日
Mask-RCNN模型的实现自定义对象(无人机)检测
计算机视觉life
17+阅读 · 2019年8月12日
Github标星4w+,如何用Python实现所有算法
七月在线实验室
5+阅读 · 2019年5月21日
Github 项目推荐 | 用 PyTorch 0.4 实现的 YoloV3
AI研习社
9+阅读 · 2018年8月11日
资源 | 深度学习图像标注工具汇总
AI100
6+阅读 · 2018年6月12日
从零开始PyTorch项目:YOLO v3目标检测实现
机器之心
14+阅读 · 2018年4月23日
深度学习界的 “吃鸡挂”——目标检测 SSD 实验
【下载】PyTorch 实现的YOLO v2目标检测算法
专知
15+阅读 · 2017年12月27日
Arxiv
5+阅读 · 2018年5月22日
Arxiv
8+阅读 · 2018年4月12日
Arxiv
4+阅读 · 2016年12月29日
VIP会员
相关资讯
开源OCR文本检测器,基于TextBoxes++和RetinaNet
专知
11+阅读 · 2019年11月15日
Mask-RCNN模型的实现自定义对象(无人机)检测
计算机视觉life
17+阅读 · 2019年8月12日
Github标星4w+,如何用Python实现所有算法
七月在线实验室
5+阅读 · 2019年5月21日
Github 项目推荐 | 用 PyTorch 0.4 实现的 YoloV3
AI研习社
9+阅读 · 2018年8月11日
资源 | 深度学习图像标注工具汇总
AI100
6+阅读 · 2018年6月12日
从零开始PyTorch项目:YOLO v3目标检测实现
机器之心
14+阅读 · 2018年4月23日
深度学习界的 “吃鸡挂”——目标检测 SSD 实验
【下载】PyTorch 实现的YOLO v2目标检测算法
专知
15+阅读 · 2017年12月27日
Top
微信扫码咨询专知VIP会员