基于pytorch的faster-rcnn目标检测框架

paper:Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
pytorch-faster-rcnn github-----pytorch-faster-rcnn

1. 简介

faster rcnn是何凯明等大神在2015年提出目标检测算法, 一直作为经典的两阶段检测算法。该算法在2015年的ILSVRV和COCO竞赛中获得多项第一。本检测框架是基于pytorch实现的faster rcnn源码。详细的论文细节请参考Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

本检测框架有如下特点:

  • 纯python代码, 完全是基于pytorch,不需要额外build
  • 个性化配置,超参数只需要在配置文件中进行配置即可进行配置
  • 有多个backbone训练网络. 包括vgg, resnet-fpn, mobilenet, high resolution net(HRNet)等
  • 只需修改配置参数即可进行不同模型的训练,对比不同超参数
  • 内存占用少,vgg16只需3G左右

2. 安装

2.1 Prerequisites

  • Python 2.7 or 3.5
  • 0
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值