Bi-box_Regression

KO | EN

gitlite — search

#javascript #python #react #golang #android #nodejs #java #windows #kubernetes #vue #security #ios #openai #open-source

Bi-box_Regression

(★ 86)

Pytorch implementation of Bi-box Regression for Pedestrian Detection and Occlusion Estimation (ECCV2018)

Bi-box_Regression 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Bi-box_Regression

Pytorch implementation of Bi-box Regression as described in [Bi-box Regression for Pedestrian Detection and Occlusion Estimation (ECCV2018)](https://cse.buffalo.edu/~jsyuan/papers/2018/Bi-box%20Regression%20for%20Pedestrian%20Detection.pdf).

![img1](https://github.com/rainofmine/Bi-box_Regression/blob/master/img/1.png)

The implementation is not just the same as the paper. We use one-stage detector RetinaNet followed by this [repo](https://github.com/yhenon/pytorch-retinanet). And some details may be different.

## Requirements

- Python3
- Pytorch0.4
- torchvision
- tensorboardX

## Installation

Install packages.

```
sudo apt-get install tk-dev python-tk
pip install cffi
pip install cython
pip install pandas
pip install tensorboardX
```

Build NMS.

```
cd Bi-box_Regression/lib
sh build.sh
```

Create folders.

```
cd Bi-box_Regression
mkdir ckpt mAP_txt summary weight
```

## Datasets
This repo is built for human detection. The popular annotation format for human detection（or pedestrian detection) contains three parts: full bbox, visible bbox and ignore regions such as [Citypersons](https://arxiv.org/pdf/1702.05693.pdf) and [Crowdhuman](https://arxiv.org/pdf/1805.00123.pdf). You should write them in CSV or TXT files.

### Annotations format
Three examples are as follows:

```
$image_path/img_1.jpg x1 y1 x2 y2 x1v y1v x2v y2v person
$image_path/img_1.jpg x1 y1 x2 y2 ignore
$image_path/img_2.jpg . . . . .
```

x1 y1 x2 y2 mean full bbox and x1v y1v x2v y2v mean visible bbox.
Images with more than one bounding box should use one row per box. Labels that we often use are 'person' or 'ignore'. When an image does not contain any bounding box, set them '.'.

### Label encoding file
A TXT file (classes.txt) is needed to map label to ID. Each line means one label name and its ID. One example is as follows:

```
person 0
```

## Pretrained Model

We use resnet18, 34, 50, 101, 152 as the backbone. You should download them and put them to `/weight`.

- resnet18: [https://download.pytorch.org/models/resnet18-5c106cde.pth](https://download.pytorch.org/models/resnet18-5c106cde.pth)
- resnet34: [https://download.pytorch.org/models/resnet34-333f7ec4.pth](https://download.pytorch.org/models/resnet34-333f7ec4.pth)
- resnet50: [https://download.pytorch.org/models/resnet50-19c8e357.pth](https://download.pytorch.org/models/resnet50-19c8e357.pth)
- resnet101: [https://download.pytorch.org/models/resnet101-5d3b4d8f.pth](https://download.pytorch.org/models/resnet101-5d3b4d8f.pth)
- resnet152: [https://download.pytorch.org/models/resnet152-b121ed2d.pth](https://download.pytorch.org/models/resnet152-b121ed2d.pth)

## Training

```
python train.py --csv_train <$path/train.txt> --csv_val <$path/val.txt> --csv_classes <$path/classes.txt> --depth <50> --pretrained resnet50-19c8e357.pth --model_name <model name to save>
```

## Reference

- [Focal Loss for Dense Object Detection](https://arxiv.org/abs/1708.02002)
- [Bi-box Regression for Pedestrian Detection and Occlusion Estimation](https://cse.buffalo.edu/~jsyuan/papers/2018/Bi-box%20Regression%20for%20Pedestrian%20Detection.pdf)

Bi-box_Regression

Bi-box_Regression 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Bi-box_Regression

Bi-box_Regression 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README