Using Fast-RCNN with a ILSVRC2012 model?

369 views
Skip to first unread message

Anu Chandra

unread,
Jun 28, 2015, 11:04:24 AM6/28/15
to caffe...@googlegroups.com
Hi,

I’m new to caffe. I was using the bvlc_reference_caffenet.caffemodel to do some basic detection. I recently came across Fast-RCNN and have got the demo working. The demo gives the option of using three models which appear to use the PASCAL VOC classes ie 20 different object categories. Is there a way to use the Fast-RCNN with the something similar to the bvlc_reference_caffenet.caffemodel which uses ILSVRC2012 1000 categories? Ideally, I want to re-write the demo code to use a ILSVRC2012 model.

Cheers

georgi angelov

unread,
Jun 28, 2015, 2:29:11 PM6/28/15
to caffe...@googlegroups.com
Bump! I'm also interested in that. The 20 classes are decent but more will definitely be better.

Sean Bell

unread,
Jul 2, 2015, 12:24:00 AM7/2/15
to caffe...@googlegroups.com
Sure, you could train one by computing your own selective search proposals, and then use the ImageNet 200 category detection challenge data.

You could probably actually just use the existing 1000-category ImageNet model to detect all 1000 classes and use some heuristic that if the max probability is low, then the box counts as background.  This will be missing bbox regression and won't do as well as a properly trained model, but at least it has the right number of categories.

Evgeny Smirnov

unread,
Jul 2, 2015, 7:38:33 AM7/2/15
to caffe...@googlegroups.com
If you want Fast-RCNN detector with 1000 ImageNet categories, you should train your own model using something like this method:

http://arxiv.org/abs/1407.5035

воскресенье, 28 июня 2015 г., 18:04:24 UTC+3 пользователь Anu Chandra написал:

X.T. Li

unread,
Sep 6, 2015, 5:28:42 AM9/6/15
to Caffe Users
Hi! I am also interested in training fast-rcnn on imagenet, did you successfully training the model?

在 2015年6月28日星期日 UTC+8下午11:04:24,gobob写道:

Rafael Ruiz Muñoz

unread,
Oct 13, 2015, 6:42:58 AM10/13/15
to Caffe Users
Did you figure out how to make it?

I've trained a lot of regular Caffemodels but I haven't got idea how to train this with boxes.
Reply all
Reply to author
Forward
0 new messages