Skip to content

Commit efc47a3

Browse files
authored
Update README.md
1 parent 76559c5 commit efc47a3

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,17 @@
22
Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
33

44
## Introduction
5+
6+
This project inherit the property of our [pytorch implementation of faster r-cnn](https://github.com/jwyang/faster-rcnn.pytorch). Hence, it also has the following unique features:
7+
8+
* **It is pure Pytorch code**. We convert all the numpy implementations to pytorch.
9+
10+
* **It supports trainig batchsize > 1**. We revise all the layers, including dataloader, rpn, roi-pooling, etc., to train with multiple images at each iteration.
11+
12+
* **It supports multiple GPUs**. We use a multiple GPU wrapper (nn.DataParallel here) to make it flexible to use one or more GPUs, as a merit of the above two features.
13+
14+
* **It supports three pooling methods**. We integrate three pooling methods: roi pooing, roi align and roi crop. Besides, we convert them to support multi-image batch training.
15+
16+
## Progress
17+
18+
Performance testing, finished soon!

0 commit comments

Comments
 (0)