Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fast rcnn #5

Open
wants to merge 24 commits into
base: master
Choose a base branch
from
Open

Fast rcnn #5

wants to merge 24 commits into from

Conversation

fmassa
Copy link
Owner

@fmassa fmassa commented Aug 23, 2015

Implements fast-rcnn detection framework. This PR includes

  • a simple and efficient lua implementation for ROIPooling based on SpatialAdaptiveMaxPooling
  • train and test code
  • an image-centric BatchProvider

Some code refactor is needed before this PR gets merged.
For the moment, there is no bounding-box regression.

fmassa added 20 commits July 9, 2015 11:57
@fmassa
Initial version of Fast-RCNN. Work in progress
456cf08
@fmassa
Add ROIDataLayer, still incomplete
d6a5caa
@fmassa
Merge branch 'master' of https://github.com/fmassa/object-detection.t…
23198fd 
…orch into fast-rcnn
@fmassa
Commiting what we have for the moment
722c433
@fmassa
Updating BatchProviderROI
58074c5
@fmassa
Basics of Fast-RCNN seems to be working
f1c256f
@fmassa
Almost working
3bf8968
@fmassa
Basic test for FRCNN
7c9ad2e
@fmassa
Fix bug in ROIPooling
5819a83
@fmassa
Merge branch 'master' of https://github.com/fmassa/object-detection.t…
f245f29 
…orch into fast-rcnn
@fmassa
Merge branch 'fast-rcnn' of https://github.com/fmassa/object-detectio…
cf614a2 
…n.torch into fast-rcnn
@fmassa
Improve training script for frcnn
bbe3dd2
@fmassa
Cleaning up and fix in test
05d8ad6
@fmassa
Add gitignore
3e10973
@fmassa
Fix unitialized memory in BatchProviderROI
ffaf0f2
@fmassa
Fix bug in BatchProviderROI
6836ca2
@fmassa
Add flip in training and test code
4ce9107
@fmassa
Force rois to be in cpu
aac63f6 
GPU single element access is much slower
@fmassa
Fix wrong batch-size
7b8b8e7
@fmassa
Modifications in the training procedure
4a46bd0 
Trying to mimic exactly Caffe's SGD
@fmassa
Copy link
Owner Author

fmassa commented Sep 2, 2015

Just a heads up on the status of this PR.
Using Caffenet, without bbox regression, this code gives a mAP of around 50.3 on Pascal VOC 2007.
Once I figure out what's holding back the ~1/2 points from original Fast-RCNN, I'll clean it and merge it.

szagoruyko and others added 4 commits September 22, 2015 12:04
@szagoruyko
fix bug in nms
4d0d46a
@fmassa
Merge pull request #8 from szagoruyko/fast-rcnn
2f3c563 
fix bug in nms
Fix BatchProviderROI incorrect super constructor invokation
154313c 
so the  BatchProviderROI inherited fields can be properly saved by torch.save
@fmassa
Merge pull request #9 from 0wu/fast-rcnn
9c3e73f 
Fix BatchProviderROI incorrect super constructor invokation
@fmassa fmassa mentioned this pull request Oct 1, 2015
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@fmassa @szagoruyko