Skip to content

Unofficially Pytorch implementation of High-level Semantic Feature Detection: A New Perspective for Pedestrian Detection

LightToYang/CSP-pedestrian-detection-in-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CSP PyTorch Implementation

Unofficially Pytorch implementation of High-level Semantic Feature Detection: A New Perspective for Pedestrian Detection

This code is only for CityPersons dataset, and only for center-position+height regression+offset regression model.

NOTE

This repo will no longer be updated. After all the experiments are done, I will open a new repo. Currently MR of CityPerson has reach 10.2%, time consuming 0.16 sec per frame(4K)

update

On Cityperson validation set

11.71 MR CSPNet-26.pth (NEW !)

12.56 MR CSPNet-89.pth

Requirement

Python, pytorch and other related libaries

GPU is needed

Usage

Compile lib

cd util
make all

Prepare CityPersons dataset as the original codes doing

  • For citypersons, we use the training set (2975 images) for training and test on the validation set (500 images), we assume that images and annotations are stored in ./data/citypersons, and the directory structure is
*DATA_PATH
	*annotations
		*anno_train.mat
		*anno_val.mat
	*images
		*train
		*val

Training & val

python trainval_torchstyle.py
python trainval_caffestyle.py

NOTE

using caffe style, you need to download additional pre-trained weight.

About

Unofficially Pytorch implementation of High-level Semantic Feature Detection: A New Perspective for Pedestrian Detection

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 55.4%
  • MATLAB 22.3%
  • Lua 7.6%
  • C++ 7.5%
  • C 5.4%
  • Cuda 1.7%
  • Makefile 0.1%