pytorch-retinanet

Object detector

This is a PyTorch implementation of the RetinaNet object detection algorithm with Focal Loss optimization.

RetinaNet in PyTorch

GitHub

997 stars
35 watching
249 forks
Language: Python
last commit: almost 6 years ago

Related projects:

Repository Description Stars
ruotianluo/pytorch-faster-rcnn An implementation of Faster R-CNN detection framework in PyTorch 1,816
unsky/retinanet A Python implementation of the RetinaNet architecture with focal loss for dense object detection. 225
ruotianluo/faster-rcnn-densecap-torch An implementation of Faster R-CNN object detection in PyTorch, modified from DenseCap. 85
xiong-zhitong/pytorch_rfcn An implementation of an object detection algorithm using PyTorch and region-based fully convolutional networks 279
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
alankbi/detecto A Python package for building and deploying computer vision models with PyTorch 614
chengyangfu/retinamask A Python-based implementation of a single-shot object detection algorithm using Mask R-CNN architecture. 339
maju116/platypus An R package for object detection and image segmentation with pre-trained models. 130
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 366
wongkinyiu/yolor An implementation of a unified network architecture for multiple object detection tasks using PyTorch 1,992
ruiminshen/yolo2-pytorch A PyTorch implementation of the YOLO (You Only Look Once) v2 object detection algorithm with flexible configuration and parallel training capabilities. 440
chenyuntc/dsod.pytorch An implementation of the Deep Supervised Object Detector from scratch using PyTorch. 70
ignacio-rocco/detectorch A PyTorch wrapper around Facebook AI Research's Detectron models for object detection and instance segmentation 557
lz118/deep-correlation-network Develops a deep neural network model for detecting salient objects in RGBT images using correlation information from other colors. 13