← 开源
ignacio-rocco

detectorch

Detectorch - detectron for PyTorch

Model DevelopmentDeep Learning FrameworksJupyter Notebook
在 GitHub 打开
增长势头
+024 小时新增 Star0.0%
558
Star
71
Fork
+0
本周
1
贡献者
创建于 2018-03-07 · 更新于 2026-10-06 · 今日第 17065 名
主要开发者
ignacio-rocco

ignacio-rocco/detectorch 排名第 17,065,有 558 个星标。

同类别比较

ignacio-rocco/detectorch 排名第 17,065,此处有 8 个同类条目;hustvl/Senna 排名第 17,064。

同类别比较
名称排名指标
DengPingFan/PraNet17,051
语言:
Python
星标数:
558
分叉数:
132
简介:
PraNet: Parallel Reverse Attention Network for Polyp Segmentation, MICCAI 2020 (Oral). Code using Jittor Framework is available.
X-DataInitiative/tick17,060
语言:
Python
星标数:
558
分叉数:
121
简介:
Module for statistical learning, with a particular emphasis on time-dependent modelling
firmai/deltapy17,062
语言:
Jupyter Notebook
星标数:
558
分叉数:
56
简介:
DeltaPy - Tabular Data Augmentation (by @firmai)
hustvl/Senna17,064
语言:
Python
星标数:
558
分叉数:
50
简介:
[IJCV 2026] Senna: Bridging Large Vision-Language Models and End-to-End Autonomous Driving
materialyzeai/megnet17,068
语言:
Jupyter Notebook
星标数:
558
分叉数:
162
简介:
Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
researchmm/STTN17,070
语言:
Jupyter Notebook
星标数:
558
分叉数:
82
简介:
[ECCV'2020] STTN: Learning Joint Spatial-Temporal Transformations for Video Inpainting
sjtu-marl/malib17,072
语言:
Python
星标数:
558
分叉数:
64
简介:
A parallel framework for population-based multi-agent reinforcement learning.
MarvinTeichmann/MultiNet17,075
语言:
Python
星标数:
557
分叉数:
243
简介:
Real-time Joint Semantic Reasoning for Autonomous Driving
关键参数与规格
关键参数与规格
参数值
语言Jupyter Notebook
类别Model Development
子类别Deep Learning Frameworks
关键词Computer vision, Object detection
星标数558
分叉数71
贡献者1
创建日期2018-03-07
更新日期2026-10-06
1 天星标数0
7 天星标数0
排名17,065
上次排名16,856
排名变化-209
7 天星标排名15,889
上次 7 天星标排名15,894
7 天星标排名变化5
总星标排名17,787
上次总星标排名17,753
总星标排名变化-34
README 摘录

Detectorch - detectron for PyTorch (Disclaimer: this is work in progress and does not feature all the functionalities of detectron. ) This code allows to use some of the Detectron models for object detection from Facebook AI Research with PyTorch. It currently supports: Fast R-CNN Faster R-CNN Mask R-CNN It supports ResNet-50/101 models with or without FPN. The pre-trained models from caffe2 can be imported and used on PyTorch. Example Mask R-CNN with ResNet-101

and FPN. Evaluation Both bounding box evaluation and instance segmentation evaluation where tested, yielding the same results as in the Detectron caffe2 models. 6 | | 36224046 | | fast_rcnn_R-50-FPN_2x | 36. 8 | | 36225249 | | e2e_faster_rcnn_R-50-C4_2x | 36. 5 | | 35857281 | | e2e_faster_rcnn_R-50-FPN_2x | 37. 9 | | 35857389 | | e2e_mask_rcnn_R-50-C4_2x | 37. 8 | 35858828 | | e2e_mask_rcnn_R-50-FPN_2x| 38. 5 | 35859007 | | e2e_mask_rcnn_R-101-FPN

_2x | 40. 4 | 35861858 | Training Training code is experimental. Installation First, clone the repo with so that you also clone the Coco API. Other required packages torchvision ( ) opencv ( ) cython ( ) matplotlib ( ) scikit-image ( ) ninja ( ) *(required for Pytorch 0. 4 only)* Additionally, you need to build the Coco API and RoIAlign layer. Compiling the Coco API If you cloned this repo with you should have also cloned the cocoapi in . Compile this with: Compiling RoIAlign The RoIAlign layer was converted from the

caffe2 version. There are two different implementations for each PyTorch version: Pytorch 0. 4: RoIAlign using ATen library (lib/cppcuda). 1: RoIAlign using TH/THC and cffi (lib/cppcuda_cffi). Needs to be compiled with: Quick Start Check the demo notebook.

常见问题

记录了多少星标和分叉?

ignacio-rocco/detectorch 有 558 个星标和 71 个分叉。

仓库排名是多少?

ignacio-rocco/detectorch 排名第 17,065。

记录的语言和类别是什么?

ignacio-rocco/detectorch 有 558 个星标,语言为 Jupyter Notebook,类别为 Model Development。

哪个仓库排名最接近?

ignacio-rocco/detectorch 与 hustvl/Senna 比较,后者排名第 17,064,有 558 个星标。

README

Detectorch - detectron for PyTorch

(Disclaimer: this is work in progress and does not feature all the functionalities of detectron. Currently only inference and evaluation are supported -- no training) (News: Now supporting FPN and ResNet-101!)

This code allows to use some of the Detectron models for object detection from Facebook AI Research with PyTorch.

It currently supports:

  • Fast R-CNN
  • Faster R-CNN
  • Mask R-CNN

It supports ResNet-50/101 models with or without FPN. The pre-trained models from caffe2 can be imported and used on PyTorch.

Example Mask R-CNN with ResNet-101 and FPN.

Evaluation

Both bounding box evaluation and instance segmentation evaluation where tested, yielding the same results as in the Detectron caffe2 models. These results below have been computed using the PyTorch code:

Model box AP mask AP model id
fast_rcnn_R-50-C4_2x 35.6 36224046
fast_rcnn_R-50-FPN_2x 36.8 36225249
e2e_faster_rcnn_R-50-C4_2x 36.5 35857281
e2e_faster_rcnn_R-50-FPN_2x 37.9 35857389
e2e_mask_rcnn_R-50-C4_2x 37.8 32.8 35858828
e2e_mask_rcnn_R-50-FPN_2x 38.6 34.5 35859007
e2e_mask_rcnn_R-101-FPN_2x 40.9 36.4 35861858

Training

Training code is experimental. See train_fast.py for training Fast R-CNN. It seems to work, but slow.

Installation

First, clone the repo with git clone --recursive https://github.com/ignacio-rocco/detectorch so that you also clone the Coco API.

The code can be used with PyTorch 0.3.1 or PyTorch 0.4 (master) under Python 3. Anaconda is recommended. Other required packages

  • torchvision (conda install torchvision -c soumith)
  • opencv (conda install -c conda-forge opencv )
  • cython (conda install cython)
  • matplotlib (conda install matplotlib)
  • scikit-image (conda install scikit-image)
  • ninja (conda install ninja) (required for Pytorch 0.4 only)

Additionally, you need to build the Coco API and RoIAlign layer. See below.

Compiling the Coco API

If you cloned this repo with git clone --recursive you should have also cloned the cocoapi in lib/cocoapi. Compile this with:

cd lib/cocoapi/PythonAPI
make install

Compiling RoIAlign

The RoIAlign layer was converted from the caffe2 version. There are two different implementations for each PyTorch version:

  • Pytorch 0.4: RoIAlign using ATen library (lib/cppcuda). Compiled JIT when loaded.
  • PyTorch 0.3.1: RoIAlign using TH/THC and cffi (lib/cppcuda_cffi). Needs to be compiled with:
cd lib/cppcuda_cffi
./make.sh 

Quick Start

Check the demo notebook.

排名相邻
同类项目