kuangliu/pytorch-retinanet

Stars
989
Rank 46,300 (Top 1.0 %)
Language
Python
Created over 7 years ago
Updated over 5 years ago

kuangliu/pytorch-retinanet

kuangliu

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

RetinaNet in PyTorch

PyTorch-RetinaNet

Train RetinaNet with Focal Loss in PyTorch.

Reference:
[1] Focal Loss for Dense Object Detection

pytorch-cifar

95.47% on CIFAR10 with PyTorch

pytorch-fpn

Feature Pyramid Networks in PyTorch

torchcv

TorchCV: a PyTorch vision library mimics ChainerCV

pytorch-ssd

Single Shot MultiBox Detector in PyTorch [deprecated]

pytorch-groupnorm

Group Normalization in PyTorch

pytorch-training-template

PyTorch model training template

pytorch-agender

Predict age & gender in one model

pytorch-yolov2

Train yolov2 with PyTorch [deprecated]

pycaffe-mtcnn

MTCNN with pycaffe

pytorch-metrics

Accuracy, precision, recall, confusion matrix computation with batch updates

mocha

Convert torch model to/from caffe model easily

pytorch-cpn

Cascaded Pyramid Network(CPN) in PyTorch

transformer

Simple transformer model for CIFAR10

pytorch-roipooling

PyTorch roipooling with torchvision

kitti-utils

KITTI coordinate transition & visualization tool

mobox

mobox: A toolbox for Motion Prediction

graph_seg

A matlab implementation of the algorithm described in the paper Efficient Graph-Based Image Segmentation.

TorchBox

PyTorch training toolbox.

kalman_filter_cpp

Kalman Filter in C++

transforms

Image & bbox transforms

tic-tac-toe

Simple RL example of game Tic-tac-toe

kalman_filter

Linear Kalman Filter

GMM

Gaussian Mixture Model via EM

yabai-config

My Yabai Skhd config for Mac

AlgorithmsSedgewick-master

FER2

LRUCache

Example usage of LRU cache

lazy.nvim

My lazy.nvim config

cmake-example

xlandmark

Facial landmark regression with xtorch

fonts

torch-cifar

Train cifar10 model with torch

dwm

easy21

RL course assignment: Easy21

vimrc

libtorch-example

Simple libtorch example

FDDB

xtorch

Torch extension for easy model training & test

lstm

Simple LSTM example

dgl