Imagenet bounding box problems?

499 views
Skip to first unread message

gobob

unread,
Jul 19, 2015, 5:13:56 PM7/19/15
to caffe...@googlegroups.com

Anyone had any issues with bounding box data downloaded from Imagenet? 


When I train a fast-rcnn model with just 2 images and their annotations for one class, it trains the model. When I extend the dataset to 14 images/annotations or bigger eg 500, I get an "assert (boxes[:, 2] >= boxes[:, 0]).all()"


Traceback (most recent call last):

  File "./tools/train_net.py", line 81, in <module>

    roidb = get_training_roidb(imdb)

  File "/home/ubuntu/fast-rcnn/tools/../lib/fast_rcnn/train.py", line 111, in get_training_roidb

    imdb.append_flipped_images()

  File "/home/ubuntu/fast-rcnn/tools/../lib/datasets/imdb.py", line 111, in append_flipped_images

    assert (boxes[:, 2] >= boxes[:, 0]).all()

AssertionError


Yohai Devir

unread,
Sep 16, 2015, 7:38:01 AM9/16/15
to Caffe Users
Many imagenet annotations are invalid.
The bounding box xmax/ymax values are sometime out of the image boundary.
Note that the values of the image width / height may not be reliable neither.
Yohai

Xueting Li

unread,
Sep 17, 2015, 10:07:14 PM9/17/15
to Caffe Users
I encountered the same problem, how can I solve this problem?

在 2015年9月16日星期三 UTC+8下午7:38:01,Yohai Devir写道:

gobob

unread,
Sep 19, 2015, 8:05:12 PM9/19/15
to Caffe Users
There is a solution for the assert issue here.

Reply all
Reply to author
Forward
0 new messages