kuangliu/pytorch-ssd

Stars
109
Rank 319,077 (Top 7 %)
Language
Python
License
MIT License
Created almost 8 years ago
Updated almost 7 years ago

kuangliu/pytorch-ssd

kuangliu

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

Single Shot MultiBox Detector in PyTorch [deprecated]

PyTorch-SSD has been deprecated. Please see torchcv, which includes an implementation of SSD300/SSD512.

Single Shot MultiBox Detector in PyTorch.

Test

Use pretrained VGG16 model

I do not recommend training SSD from scratch. Use pretrained VGG model helps a lot to achieve lower losses.

I use the pretrained pytorch/vision VGG16 model from PyTorch model zoo.

Credit

This implementation is initially inspired by:

pytorch-cifar

95.47% on CIFAR10 with PyTorch

pytorch-retinanet

RetinaNet in PyTorch

pytorch-fpn

Feature Pyramid Networks in PyTorch

torchcv

TorchCV: a PyTorch vision library mimics ChainerCV

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