Skip to content

nathanjjohnson7/rpn-for-faster_rcnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

rpn-for-faster_rcnn

Region Proposal Network Implementation for Faster RCNN

The RPN used to create outputs below was trained for only 40000 epochs instead of 80000.

alt text

alt text

alt text

alt text

alt text

alt text

The Faster RCNN is a combination of the Region Proposal Network and the Fast RCNN.

To build a Faster RCNN model using 4-stage-training:

1.train the RPN (rpn-for-faster_rcnn/build_rpn.py)

2.get proposals from RPN (rpn-for-faster_rcnn/create_rpn_dataset.py)

3.train Fast RCNN with dataset (fast_rcnn/build_classifier.py)

4.train another RPN initialized with frozen base model weights from Fast RCNN

5.get proposals from new RPN

6.continue training Fast RCNN with new dataset (keep base model weights frozen)

About

Region Proposal Network Implementation for Faster RCNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages