上传者: 42109639
|
上传时间: 2021-10-28 21:22:29
|
文件大小: 297KB
|
文件类型: -
pytorch-faster-rcnn
1.简介
基于Pytorch的快速rcnn框架的实现。有关更快的R-CNN的详细信息,请参阅论文《 ,作者邵少青,何开明,Ross Girshick,孙健
此检测框架具有以下功能:
它可以作为纯python代码运行,也可以基于pytorch框架纯运行,无需构建
仅运行train.py脚本即可轻松进行培训,只需设置数据根目录
它有许多骨干网。 像vgg,resnet-fpn,mobilenet,高分辨率网络(HRNet)
它可以是一个真正的检测框架。 您只需要在配置文件中更改超级参数,并获得不同的模型来比较不同的模型
它的内存效率高(vgg16约为3GB)
2.安装
2.1先决条件
Python 2.7或3.5
火炬1.5.1
火炬视觉0.6.1
numpy的1.15.4
枕头6.1.0
pycocotools 2.0
matplotl