background class

[scott-vsi]

I ran the demo.py script with the provided ycb_prototypes.pth prototype model file. However, I only used the first 10 prototypes for both label_names and test_class_weight (apple, ball, banana, bowl, brick, can, cheez-it, chips, clamp, cleanser bottle).

The labeled ycb.jpg image that is produced contains a lot of bad detections, like labeling the toy airplane, 'cheez-it'.

Is this expected? Or am I doing something wrong?

EDIT Or is the the problem that this item is technically a foreground class, so it is unlikely under the background class model; as a result, it gets labeled as some foreground class.