leeyoshinari / YOLO_v2

The implementation of YOLO v2 with TensorFlow
GNU General Public License v3.0
80 stars 38 forks source link

Is it assumed that there is only one object in each cell? #34

Open rogercw opened 5 years ago

rogercw commented 5 years ago

Thanks for sharing the codes. I am little confused about the way to generate the ground truth labels. Hope you could help me point out what I might misunderstand. Thanks.

When generating training labels in https://github.com/leeyoshinari/YOLO_v2/blob/aa9bf78ecf461a1a3924e466024f8417d23197aa/pascal_voc.py#L61-L76, won't the labels be overwritten if two objects in the same cell? If that is the case, does that mean the first object can not be used for training?

yuanliangxie commented 5 years ago

YOLOv2假定了每个cell至多含有一个grounth truth,而在实际上基本不会出现多于1个的情况。